你这个问题信息太少了,我没法直接判断“为何报错”。在 Linux + PyTorch 环境下,常见报错原因大致有下面几类,你可以对照看看,也可以把完整报错信息贴出来,我帮你精确定位。
报错示例:
CUDA error: no kernel image is available
RuntimeError: CUDA out of memory
原因:
排查:
nvidia-smi
nvcc --version
python -c "import torch; print(torch.version.cuda)"
报错示例:
ImportError: libcuda.so.1: cannot open shared object file
原因:
解决:
pip uninstall torch
pip install torch --index-url https://download.pytorch.org/whl/cu118
报错示例:
GLIBCXX_3.4.29 not found
原因:
解决:
/tmp 无写权限model.to(device) 没用请贴出:
python - <<'PY'
import torch
print(torch.__version__)
print(torch.cuda.is_available())
PY
uname -a
nvidia-smi
如果你愿意,可以直接把报错截图或文字发上来,我一步一步帮你修。