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

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

系统更新与依赖安装

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

创建虚拟环境（可选）

安装virtualenv：pip3 install virtualenv。创建并激活虚拟环境：virtualenv pytorch_env，source 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 torch，print(torch.__version__)和print(torch.cuda.is_available())，若版本号正确且cuda.is_available()返回True（使用GPU时），则安装成功。

安装其他库（可选）

根据项目需求，用pip安装如pandas、matplotlib等库。