PyTorch在CentOS上的使用教程是什么

centos

小樊

2025-03-01 19:15:25

栏目: 智能运维

在CentOS上安装和使用PyTorch可以分为几个步骤，包括安装Anaconda、创建虚拟环境、安装PyTorch以及验证安装。以下是详细的教程：

安装Anaconda

首先，你需要在CentOS上安装Anaconda。你可以参考这个教程来了解如何在CentOS上安装Anaconda。

创建虚拟环境

在安装PyTorch之前，建议创建一个虚拟环境来隔离你的项目环境。你可以使用以下命令来创建一个名为study_torch的虚拟环境，并指定Python版本为3.10：

conda create -n study_torch python=3.10

安装PyTorch

在虚拟环境中，你可以通过以下命令来安装PyTorch。请根据你的CUDA版本选择合适的PyTorch版本。以下是一个示例命令，用于安装CUDA 12.1版本的PyTorch：

conda install pytorch torchvision torchaudio cudatoolkit=12.1 -c pytorch

如果你不想安装CUDA，可以选择CPU版本的PyTorch：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

你还可以使用清华大学的镜像源来加速下载：

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
pip install torch torchvision torchaudio

验证安装

为了验证PyTorch是否安装成功，你可以打开Python解释器并尝试导入PyTorch：

import torch
x = torch.rand(5, 3)
print(x)

如果没有报错并且输出了类似于tensor([[0.3089, 0.5494, 0.1153], [0.9371, 0.8981, 0.3798], [0.3220, 0.4789, 0.6379], [0.3280, 0.0454, 0.5896], [0.2760, 0.0586, 0.2910]])的输出，那么说明PyTorch已经成功安装。

配置环境变量（可选）

为了方便使用，你可以将Python和PyTorch的路径添加到系统的环境变量中。编辑~/.bash_profile文件，添加以下内容：

export PYTHON_HOME=/path/to/your/python
export PATH=$PYTHON_HOME/bin:$PATH

然后执行以下命令使环境变量生效：

source ~/.bash_profile

安装Jupyter Notebook（可选）

如果你需要使用Jupyter Notebook进行开发，可以通过以下命令安装：

conda install jupyter
jupyter notebook --no-browser --ip 0.0.0.0 --port 7777 &

这样，你就可以在浏览器中通过http://your_server_ip:7777访问Jupyter Notebook了。

以上就是在CentOS上安装和使用PyTorch的基本教程。希望对你有所帮助！

亿速云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>