Ubuntu PyTorch常见问题及解决方案

在Ubuntu系统上安装和配置PyTorch时，用户可能会遇到一些常见问题。以下是一些常见问题及其解决方案：

常见问题及解决方案

• ImportError: libmkl_intel_lp64.so: cannot open shared object file

  解决办法：在.bashrc中加入一行 export LD_LIBRARY_PATH=/root/anaconda3/lib:$LD_LIBRARY_PATH，然后运行 source ~/.bashrc 更新环境变量。

• ModuleNotFoundError: No module named ‘torch’

  解决办法：确保已经安装了PyTorch。可以使用 conda install pytorch torchvision torchaudio cudatoolkit=X.X -c pytorch 命令安装，其中X.X是CUDA版本。

• RuntimeError: set_sizes_contiguous is not allowed on a Tensor created from .data or .detach()

  解决办法：参考报错提示进行修改，去掉 .data。

• CUDA版本不匹配

  解决办法：检查 CUDA_HOME 环境变量是否正确设置，并确保安装的PyTorch版本与CUDA版本匹配。

• 编译报错

  解决办法：确保使用的CUDA和cuDNN版本与PyTorch兼容。可以参考PyTorch官网的安装指南，选择合适的版本进行安装。

• 网络问题导致无法安装PyTorch

  解决办法：尝试更换国内镜像源，如清华大学提供的源。如果仍然失败，可以手动下载whl文件进行本地安装。

• No module named ‘torch._C’

  解决办法：尝试重命名 _C.cpython-35m 等文件为 _C.so，然后重新导入。

安装步骤

1. 更新系统包列表

   sudo apt update
   

2. 安装必要的系统依赖

   sudo apt install -y build-essential cmake git wget unzip yasm pkg-config libopenblas-dev liblapack-dev libjpeg-dev libpng-dev
   

3. 安装Python和pip

   sudo apt install -y python3 python3-pip
   

4. 创建虚拟环境（可选）

   python3 -m venv pytorch-env
   source pytorch-env/bin/activate
   

5. 安装PyTorch

   使用pip安装PyTorch：

   pip install torch torchvision torchaudio
   

   如果需要安装特定版本的PyTorch，可以参考PyTorch官网获取安装命令。

6. 验证安装

   在Python环境中运行以下命令，检查PyTorch是否安装成功：

   import torch
   print(torch.__version__)
   print(torch.cuda.is_available())
   

   如果安装了GPU版本，应该返回 True。