Ubuntu 18.04安装NVIDIA Driver+CUDA 11.4.4+ cuDNN 8.2.2

Ubuntu 18.04安装NVIDIA Driver+CUDA 11.4.4+ cuDNN 8.2.2

下载安装对应的CUDA与cuDNN

CUDA下载与安装

CUDA下载： 地址

下载CUDA前需要在终端输入：ubuntu-drivers devices 查看显卡支持的驱动版本
显示如下：

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00002482sv00007377sd00001581bc03sc00i00
vendor   : NVIDIA Corporation
manual_install: True
driver   : nvidia-driver-510 - distro non-free
driver   : nvidia-driver-515-server - distro non-free
driver   : nvidia-driver-520 - distro non-free
driver   : nvidia-driver-470-server - distro non-free
driver   : nvidia-driver-510-server - distro non-free
driver   : nvidia-driver-470 - distro non-free recommended
driver   : nvidia-driver-515 - distro non-free
driver   : xserver-xorg-video-nouveau - distro free builtin

说明该显卡可安装470、515、510、520等版本的驱动，并查找支持这些版本驱动的CUDA，比如我使用的版本为470，可选择CUDA版本如下：然后通过安装指导下载并安装

wget https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.run
sudo sh cuda_11.4.4_470.82.01_linux.run

安装过程中弹出对话框选择默认安装(同时安装显卡驱动与CUDA)
安装完成之后添加变量环境：
在终端输入：gedit ~/.bashrc，并在文件中最后一行添加：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.4/lib64
export PATH=$PATH:/usr/local/cuda-11.4/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.4

CUDNN下载与安装

cuDNN下载：​​​​​​地址
在下载CUDNN时注意要与CUDA版本号对应，我安装的是8.2.2，
解压下载的文件，找到include/lib64所在的文件夹，复制文件夹绝对路径替换下面命令中的XXX并执行命令：

    sudo cp XXX/include/cudnn.h /usr/local/cuda/include/
    sudo cp XXX/lib64/libcudnn* /usr/local/cuda/lib64/
    sudo chmod a+r /usr/local/cuda/include/cudnn.h
   sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

最后sudo reboot重启，并通过如下命令查看显卡驱动、CUDA、CUDNN 是否安装成功：
nvidia-smi 查看显卡驱动
nvcc --version 查看CUDA

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 查看CUDNN(不适用于8.0以上版本CUDNN)

完成安装

卸载CUDA11.4

执行命令：

cd /usr/local/cuda-11.4/bin

再执行：

sudo ./cuda-uninstaller

在弹出的对话框中全部勾选，然后选择done按下回车键，等待一段时间后，提示Successfully uninstalled 完成卸载。

显卡驱动失效处理

安装一段时间后，当再次打开系统时发现显卡驱动失效了。
解决办法：
在终端输入nvidia-smi，显示如下：

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

在输入nvcc -V检查cuda，显示

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Mon_Oct_11_21:27:02_PDT_2021
Cuda compilation tools, release 11.4, V11.4.152
Build cuda_11.4.r11.4/compiler.30521435_0

则说明驱动存在。
接下来，执行以下命令，查看已安装驱动的版本信息。

ls /usr/src | grep nvidia

比如我的显示： nvidia-470.82.01；则在终端执行如下命令

sudo apt-get install dkms
sudo dkms install -m nvidia -v 470.82.01

再次nvidia-smi查看安装结果。

reference:
https://ai-wx.blog.csdn.net/article/details/108997692?spm=1001.2014.3001.5506