CentOS如何配置PyTorch环境

以下是在CentOS上配置PyTorch环境的步骤：

1. 更新系统：sudo yum update -y。
2. 安装依赖包：sudo yum install -y python3 python3-pip python3-devel，若用CUDA还需安装相关依赖。
3. 创建虚拟环境（可选）：python3 -m venv pytorch_env，然后source pytorch_env/bin/activate。
4. 安装PyTorch：
   • CPU版本：pip3 install torch torchvision torchaudio。
   • GPU版本：先安装CUDA和cuDNN，再根据版本选择命令，如CUDA 11.7用pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117。
5. 验证安装：在Python解释器中输入import torch，然后print(torch.__version__)和print(torch.cuda.is_available())。