解决使用pytorch过程中提示的“The NVIDIA driver is too old”问题

目录

 

问题详情

原因

解决办法

---

问题详情

The NVIDIA driver on your system is too old (found version xxxx).Please update your GPU driver by downloading and installing a new version from the URL: Download Drivers
Alternatively, go to: PyTorch to install a PyTorch version that has been compiled with your version of the CUDA driver.

原因

项目环境中的pytorch版本与系统的nvidia显卡驱动版本不匹配。

1、查看项目环境的pytorch版本：

#python环境下
import torch
print(torch.__version__)

2、查看CUDA版本信息以及显卡驱动版本信息：

#terminal

#cuda
nvcc --version
#nvidia driver
nvidia-smi

3、pytorch、cuda、nvidia driver版本匹配关系

pytorch版本与cuda的匹配关系：https://pytorch.org/get-started/previous-versions/

cuda版本与nvidia driver的匹配关系：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

解决办法

重新安装nvidia driver。

1、在nvidia官网根据显卡型号及平台搜索驱动：https://www.nvidia.cn/geforce/drivers/ ，在下方的驱动程序搜索结果中选择相应的驱动进行下载。

2、安装nvidia driver

卸载原有驱动：

sudo apt-get remove --purge nvidia*

临时禁用图形界面：

sudo service lightdm stop

按下Ctrl+Alt+F1进入terminal，输入账户信息进行登录。

进入存放驱动文件的文件夹，进行安装。

#为所有用户增加执行权限
sudo chmod a+x NVIDIA-Linux-x86_64-450.80.02.run
#执行安装nvidia driver操作
#-no-x-check: 安装驱动时关闭x服务
#-no-nouveau-check: 安装驱动时禁用Nouveau
#-no-opengl-files: 安装时只装驱动文件，不安装Opengl
sudo ./NVIDIA-Linux-x86_64-450.80.02.run -no-x-check -no-nouveau-check -no-opengl-files

在安装界面中按照默认选项执行，安装完成后查看是否安装成功。

nvidia-smi

开启图形界面服务并重启：

sudo service lightdm start
#重启或者同时按Ctrl+Alt+F7返回图形界面
reboot