ubuntu服务器版NVIDIA驱动失效解决方案

1. 问题描述

在终端输入nvidia-smi，输出如下：

NVIDIA-SMI has failed because it couldn't communicate with the NvIDIA driver, Make sure that the latest NIDIA driveris installed and running.

输入nvcc -V,输出如下：

nvcc:NVIDIA(R)Cuda compiler driver
Copyright(c)2005-2022 NVIDIA Corporation
Built on Mon oct 24 19:12:58 PDT 2022
Cuda compilation tools,release 12.0,v12.0.76
Build cuda 12.0.r12.0/compiler.31968024 0

如下图所示：

这种情况应该是ubuntu内核和显卡驱动不匹配，网上的教程包括降低ubuntu内核版本（参考Ubuntu 修改默认启动内核），NVIDIA驱动失效简单解决方案（参考NVIDIA驱动失效简单解决方案：NVIDIA-SMI has failed because it couldn‘t communicate with the NVIDIA driver.）,亲测，对本人问题失效，但是大家遇到这种问题可以先试试这两个，因为操作简单。

2. 解决方法–卸载并重新安装最新版显卡驱动+cuda+cudnn

2.1 卸载显卡驱动

卸载驱动

sudo apt-get --purge remove nvidia*
sudo apt autoremove
sudo apt-get --purge remove "*cublas*" "cuda*"
sudo apt-get --purge remove "*nvidia*"

以下命令查看显卡驱动是否卸载干净，没有输出表示卸载完成。

dpkg -l | grep nvidia 

2.2 重新安装最新版显卡驱动+cuda+cudnn

2.2.1 显卡驱动

在NVIDIA官网搜索最新显卡驱动，下载，上传到服务器根目录。

这是我的显卡目前最新版本的驱动。

然后安装显卡（这里是卸载重装的简单代码，若第一次安装显卡驱动，需要进行一些配置，参考ubuntu22.04安装显卡驱动+cuda+cudnn）：

sudo chmod a+x NVIDIA-Linux-x86_64-560.35.03.run 

sudo ./NVIDIA-Linux-x86_64-560.35.03.run -no-x-check -no-nouveau-check -no-opengl-files

运行中红色选项参考ubuntu22.04安装显卡驱动+cuda+cudnn

安装完成运行nvidia-smi，出现如下所示，表明运行成功。

2.2.2 cuda

从上面的图中可以看到显卡最高可以支持cuda12.6

网址：https://developer.nvidia.com/cuda-toolkit-archive

选择上图红色标记的cuda，点击CUDA Toolkit 12.6.0，以下选择。

出现如下所示提示安装命令，

以上前三行是有关下载.deb文件的命令，参考我的上一篇博客ubuntu22.04安装显卡驱动+cuda+cudnn，可以使用以下命令替代：

axel -n 50 https://developer.download.nvidia.com/compute/cuda/12.6.0/local_installers/cuda-repo-ubuntu2204-12-6-local_12.6.0-560.28.03-1_amd64.deb

然后执行第四行和第四行之后代码：

sudo dpkg -i cuda-repo-ubuntu2204-12-6-local_12.6.0-560.28.03-1_amd64.deb
# 注意下面这行*根据上一个命令输出结果提示修改
sudo cp /var/cuda-repo-ubuntu2204-12-6-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-6

接下来需要配置环境变量，

1. 打开.bashrc文件；
2. 在文件末尾删除之前的cuda配置命令，例如，

export PATH=/usr/local/cuda-12.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

3. 在文件末尾输入如下内容并保存退出；

export PATH=/usr/local/cuda-12.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

4. 激活

source ~/.bashrc

5. 检查cuda是否安装成功，下图表示安装成功。
   

2.2.3 cuda安装cudnn

卸载cudnn参考CUDA与cuDNN在linux / Ubuntu22.04上的安装与卸载，包含CUDA的.run安装与.deb安装，cuDNN的.tar安装与.deb安装，其中

sudo dpkg -r cudnn-local-repo-ubuntu2204-8.9.7.29
sudo rm -rf /var/cudnn-local-repo-ubuntu2204-8.9.7.29

和

sudo dpkg --purge cudnn-local-repo-ubuntu2204-8.9.7.29

很有用。

安装cudnn和测试参考ubuntu22.04安装显卡驱动+cuda+cudnn。