在 Ubuntu 上运行 PyTorch 报错的原因非常多,不能一概而论,需要看具体报错信息。下面我按常见场景 + 典型报错 + 解决方法给你梳理一下,你可以对照看是哪一种。
报错示例
CUDA error: no kernel image is available
或
torch.cuda.is_available() → False
原因
解决
nvidia-smi # 看驱动和CUDA
然后按官网安装对应版本: https://pytorch.org/get-started/locally/
如果没有 GPU:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
报错示例
ImportError: libtorch_cpu.so: cannot open shared object file
或
ModuleNotFoundError: No module named 'torch'
原因
解决
which python
pip show torch
推荐用 conda:
conda create -n pt python=3.10
conda activate pt
pip install torch
报错示例
version `GLIBCXX_3.4.30' not found
原因
解决
pip install torch==1.13.1
报错示例
libgomp.so.1: cannot open shared object file
解决
sudo apt update
sudo apt install libgomp1
现象
解决
python -m ipykernel install --user --name pt
然后在 Jupyter 选对应 kernel
请把下面信息发我,我可以直接帮你判断:
lsb_release -a)nvidia-smi)sudo apt install python3-venv
python3 -m venv pt
source pt/bin/activate
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
如果你愿意,把具体报错复制出来,我可以直接告诉你“这一行是啥意思、怎么修”。