Ubuntu如何解决PyTorch安装问题

以下是Ubuntu解决PyTorch安装问题的关键步骤：

1. 检查系统环境

   • 确保Ubuntu版本符合PyTorch要求，安装Python 3.6+和pip。
   • 若需GPU支持，安装对应版本的CUDA和cuDNN，并验证驱动是否正确。

2. 选择安装方式

   • 推荐Conda：创建虚拟环境后安装，自动处理依赖（如CUDA版本匹配）。

     conda create -n pytorch_env python=3.8  
     conda activate pytorch_env  
     conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c nvidia  # 替换为对应CUDA版本  
     

   • pip安装：需手动指定CUDA版本，适合无Conda的场景。

     pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117  # 替换为对应CUDA版本  
     

3. 解决常见问题

   • 版本不匹配：确保PyTorch、CUDA、cuDNN版本兼容，可尝试降级版本。
   • 网络问题：使用清华源加速下载（需在安装命令前配置镜像源）。
   • 权限问题：若安装失败，尝试添加执行权限或使用sudo（谨慎使用）。
   • 依赖缺失：安装编译工具和库（如build-essential、libopenblas-dev）。

4. 验证安装
   运行以下代码检查是否成功：

   import torch  
   print(torch.__version__)  
   print(torch.cuda.is_available())  # 若为True则GPU支持正常  
   

参考来源：