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，然后重新导入。

安装必要的系统依赖

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

安装Python和pip

sudo apt install -y python3 python3-pip

创建虚拟环境（可选）

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

安装PyTorch

使用pip安装PyTorch：
```
pip install torch torchvision torchaudio
```
如果需要安装特定版本的PyTorch，可以参考PyTorch官网获取安装命令。
验证安装

在Python环境中运行以下命令，检查PyTorch是否安装成功：
```
import torch
print(torch.__version__)
print(torch.cuda.is_available())
```
如果安装了GPU版本，应该返回 True。

0 赞

0 踩