在 Ubuntu 上运行 PyTorch 报错的原因有很多,常见情况如下,你可以对照排查:
报错示例:
CUDA error: no kernel image is available
Found no NVIDIA driver on your system
原因:
.cuda()解决:
nvidia-smi # 看驱动和 CUDA 版本
nvcc --version # 看编译环境
安装对应版本 PyTorch:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
报错示例:
ModuleNotFoundError: No module named 'torch'
原因:
解决:
which python
pip show torch
确保安装和运行是同一个环境。
报错示例:
PermissionError
OSError: libcudnn.so not found
解决:
sudo pipsudo apt install libgl1 libglib2.0-0
报错示例:
ImportError: /lib/x86_64-linux-gnu/libc.so.6: version `GLIBC_2.31' not found
原因:
解决:
报错示例:
Illegal instruction (core dumped)
原因:
解决:
lscpu | grep avx
用老版本 PyTorch(如 1.12 CPU 版)
请贴出:
python -Vpip show torchnvidia-smi(如有 GPU)我可以精确告诉你哪里错了。