PyTorch无法调用CUDA:Error 803驱动版本不兼容问题求助
解决Ubuntu 22.04下PyTorch无法识别CUDA(Error 803)的问题
问题根源
Error 803本质是NVIDIA驱动与CUDA版本不兼容,或是驱动未适配当前系统内核(你使用的6.1.15-generic属于非默认内核,可能存在驱动适配问题)。CUDA 11.7要求驱动版本≥450.80.02,需先确认当前驱动是否满足要求。
解决步骤
1. 检查当前驱动与CUDA版本
执行以下命令获取关键信息:
# 查看驱动版本及驱动支持的最高CUDA版本 nvidia-smi # 查看系统安装的CUDA编译工具版本 nvcc --version
- 若
nvidia-smi显示的Driver Version低于450.80.02,直接进入步骤2升级驱动。 - 若
nvcc --version显示的CUDA版本不是11.7,需先确保系统CUDA 11.7安装正确。
2. 安装适配6.1内核的NVIDIA驱动
由于你使用的是6.1.x内核,Ubuntu默认仓库的驱动可能版本偏低,需通过官方PPA安装适配版本:
# 添加NVIDIA图形驱动PPA sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt update # 安装对应CUDA 11.7的兼容驱动(515系列适配6.1内核且支持CUDA 11.7) sudo apt install linux-headers-$(uname -r) nvidia-driver-515 # 重启系统使驱动生效 sudo reboot
3. 验证驱动与CUDA兼容性
重启后再次执行nvidia-smi,确认:
- Driver Version为515.x及以上
- 右上角CUDA Version≥11.7
同时nvcc --version需显示release 11.7。
4. 重新安装PyTorch(可选,若前序步骤后仍报错)
若conda安装的PyTorch与系统CUDA存在冲突,可卸载后改用pip安装对应版本:
# 卸载conda环境中的PyTorch相关包 conda uninstall pytorch torchvision torchaudio pytorch-cuda -y # 用pip安装适配CUDA 11.7的PyTorch 1.13.1 pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
5. 测试CUDA可用性
运行以下Python代码验证:
import torch print("CUDA Available:", torch.cuda.is_available()) print("PyTorch CUDA Version:", torch.version.cuda) print("GPU Device Count:", torch.cuda.device_count())
若输出CUDA Available: True则问题解决。
内容的提问来源于stack exchange,提问作者Rotkiv
相关产品推荐
相关产品推荐

