如何在CentOS上解决PyTorch的依赖问题

在CentOS上解决PyTorch依赖问题可按以下步骤操作：

1. 更新系统：

   sudo yum update -y  
   

2. 安装基础依赖：

   sudo yum install -y python3 python3-pip gcc make cmake3 git wget  
   

3. 配置CUDA环境（GPU支持需）：

   • 下载并安装CUDA Toolkit（需与PyTorch版本兼容），添加环境变量：

     export PATH=/usr/local/cuda/bin:$PATH  
     export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH  
     

   • 安装对应版本的cuDNN（从NVIDIA官网下载）。

4. 安装PyTorch：

   • CPU版本：

     pip3 install torch torchvision torchaudio  
     

   • GPU版本：

     pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu11x  # 替换为实际CUDA版本（如cu117）  
     

   • 推荐使用conda（避免依赖冲突）：

     conda create -n pytorch_env python=3.8  
     conda activate pytorch_env  
     conda install pytorch torchvision torchaudio cudatoolkit=11.x -c pytorch  
     

5. 验证安装：

   import torch  
   print(torch.__version__)  
   print(torch.cuda.is_available())  # GPU版本应返回True  
   

常见问题解决：

• 依赖冲突：升级pip/setuptools，或手动安装冲突包。
• 网络慢：使用国内镜像源（如清华源）。
• 版本不兼容：确保CUDA、cuDNN与PyTorch版本匹配。

参考来源：