“Linux 下 PyTorch 报错”本身是一个很宽泛的问题,报错原因非常多,需要看具体报错信息才能准确判断。不过我可以先给你最常见的几类原因 + 排查方法,你对照看看自己属于哪一种。
典型报错:
CUDA unavailable
AssertionError: Torch not compiled with CUDA enabled
✅ 原因:
✅ 解决:
nvidia-smi # 看驱动是否识别
nvcc --version # 看 CUDA 是否安装
python -c "import torch; print(torch.cuda.is_available())"
如果是 False:
pip uninstall torch
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
典型报错:
GLIBCXX not found
undefined symbol
Segmentation fault
✅ 原因:
✅ 建议:
conda create -n torch python=3.9
conda activate torch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
典型报错:
libcudnn.so not found
libgomp.so.1: cannot open shared object file
✅ 解决:
sudo apt install libgomp1
sudo apt install libcudnn8
或 conda:
conda install -c conda-forge libgomp
典型报错:
Permission denied
No module named torch
✅ 检查:
which python
pip show torch
✅ 建议:
sudo pip典型报错:
size mismatch
Missing key(s) in state_dict
✅ 原因:
你可以把下面内容贴出来:
python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"
cat /etc/os-release
nvidia-smi
如果你愿意,可以直接把报错截图或复制文本发给我,我可以逐行帮你分析。