温馨提示×

在CentOS上如何搭建PyTorch开发环境

小樊
48
2025-08-15 00:14:50
栏目: 智能运维

以下是在CentOS上搭建PyTorch开发环境的步骤:

系统更新与依赖安装

更新系统:sudo yum update -y。安装依赖项:sudo yum groupinstall -y "Development Tools" sudo yum install -y python3 python3-devel numpy

创建虚拟环境(可选)

安装virtualenvpip3 install virtualenv。创建并激活虚拟环境:virtualenv pytorch_envsource pytorch_env/bin/activate

安装PyTorch

  • 使用pip安装:访问PyTorch官网,根据CUDA版本选择安装命令,如CUDA 11.7:pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117;若不需要CUDA,直接用pip3 install torch torchvision torchaudio
  • 使用Anaconda安装:先安装Anaconda,然后创建虚拟环境conda create -n pytorch_env python=3.10,激活后用conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch安装对应CUDA版本的PyTorch。

验证安装

在Python中运行import torchprint(torch.__version__)print(torch.cuda.is_available()),若版本号正确且cuda.is_available()返回True(使用GPU时),则安装成功。

安装其他库(可选)

根据项目需求,用pip安装如pandasmatplotlib等库。

0