在 Ubuntu 上运行 PyTorch 报错的原因非常多,没有具体报错信息很难精准判断。下面我按「最常见原因 + 对应解决办法」给你一个排查清单,你可以对照看是哪一种。
典型报错:
CUDA driver version is insufficientno kernel image is availableRuntimeError: CUDA error✅ 解决思路:
nvidia-smi
例如:
nvidia-smi 显示 CUDA 11.8torch 的 CUDA 12.1 → 会报错✅ 正确安装方式(示例):
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
典型报错:
AssertionError: Torch not compiled with CUDA enableddevice(type='cuda') 报错✅ 检查:
import torch
print(torch.cuda.is_available())
如果是 False,说明没装 CUDA 版 PyTorch
✅ 重新安装 GPU 版(见上面)
典型现象:
ModuleNotFoundError: No module named 'torch'✅ 检查:
which python
which pip
确保一致
✅ 建议用虚拟环境:
python -m venv venv
source venv/bin/activate
pip install torch
典型报错:
version 'GLIBCXX_3.4.30' not foundImportError: libstdc++.so.6✅ 解决:
strings /usr/lib/x86_64-linux-gnu/libstdc++.so.6 | grep GLIBCXX
升级 GCC:
sudo apt install gcc-11 g++-11
典型报错:
libcuda.so.1: cannot open shared object file✅ 解决:
sudo apt install nvidia-cuda-toolkit
或重装 NVIDIA 驱动
✅ 检查:
import sys
print(sys.executable)
请把下面信息发我(越多越好):
nvidia-smi
python -c "import torch; print(torch.__version__, torch.cuda.is_available())"
lsb_release -a)如果你愿意,可以直接把报错截图或复制文本贴出来,我一步一步帮你修。