ubuntu20.04安装nvidia驱动+cuda11.8+cudnn8.6

显卡安装

显卡驱动安装方法就不说了，很简单。若果出现问题，可以参考下文解决办法
启动 Ubuntu 系统时不让系统加载图形化界面，在终端命令模式下卸载 NVIDIA 驱动，再重装新的驱动

显卡驱动卸载命令位置/usr/bin/nvidia-uninstall，以下命令即可卸载：

sudo /usr/bin/nvidia-uninstall

以下命令备用：

sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo modprobe -r nvidia-drm
cd NVIDIA驱动安装文件所在的目录
sudo sh ./NVIDIA驱动安装文件.run
sudo systemctl set-default graphical.target
sudo reboot 0

通过ctrl+alt+F1进入命令行界面，关闭所有使用GPU的进程（没试过！！！）。

sudo systemctl isolate multi-user.target

CUDA安装

安装好显卡驱动后，在nvidia.cn官网上下载“cuda_11.7”，用命令行进行安装：

chmod +x cuda_11.8.0_520.61.05_linux.run
sudo ./cuda_11.8.0_520.61.05_linux.run

按照提示即可完成，一般不选择驱动项目（用enter将x去掉），以免出现错误。

cudnn安装

首先下载“cudnn-local-repo-ubuntu2004-8.6.0.163_1.0-1_amd64.deb”，用以下命令进行安装：

sudo dpkg -i cudnn-local-repo-ubuntu2004-8.6.0.163_1.0-1_amd64.deb

完成安装后，其路径在如下目录下：
/var/cudnn-local-repo-ubuntu2004-8.6.0.163

接着，复制GPG到如下目录：

sudo cp /var/cudnn-local-repo-*/cudnn-local-*-keyring.gpg /usr/share/keyrings/

采用以下命令查看cudnn版本：

sudo apt-cache madison libcudnn8

将查的信息加入下面命令，完成cudnn的最终安装：

sudo apt-get install libcudnn8=8.6.0.163-1+cuda11.8
sudo apt-get install libcudnn8-dev=8.6.0.163-1+cuda11.8
sudo apt-get install libcudnn8-samples=8.6.0.163-1+cuda11.8

测试

输入以下命令：

cp -r /usr/src/cudnn_samples_v8/ $HOME
cd  $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN

出现Test Passed! 代表cudnn可以正常使用。

如果出错，说明FreeImage工具没安装，输入：

sudo apt-get install libfreeimage3 libfreeimage-dev

完成后即可测试。