常见错误汇总
CUDA版本不兼容 ⚠️
请检查NVIDIA驱动与PyTorch版本的匹配性,查看官方支持列表环境变量配置错误 🛠️
确认PATH
和LD_LIBRARY_PATH
是否包含正确路径,示例:export PATH="/usr/local/cuda/bin:$PATH" export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
网络连接超时 🌐
尝试使用镜像源或离线安装包,参考国内镜像配置教程Python环境冲突 🐍
使用虚拟环境(如venv
或conda
)隔离依赖,查看环境管理指南
解决方案步骤
- 清理缓存:
pip cache purge
- 重新安装依赖:
pip install --force-reinstall torch
- 检查系统日志:
journalctl -u nvidia-smi
(Linux) - 使用官方安装脚本:
curl -L https://pytorch.org/whl/torch_stable.html | grep -i "pip install"