Ubuntu20.04 安装 cudatookit 12.2 + cudnn 安装

最简约的部署Ubuntu20.04深度学习环境的教程

1. 安装Ubuntu20.04 系统

B站详细的安装教程
简约安装版

2. 安装Nvidia显卡驱动

我参考了各种资料，重装系统，完美解决开机显示器黑屏无法进入桌面的情况

黑屏问题主要是由linux内核更新导致，nvidia 不兼容 ubuntu的图像显示，所以很容易出现黑屏情况

在安装nvidia驱动前一定要使用命令：

# 一定要保持当前的内核，也就是安装 NVIDIA 驱动时用的内核
sudo apt-mark hold linux-image-generic linux-headers-generic

即保持当前系统的内核不更新之后，再去安装 NVIDIA 驱动，成功解决问题

打开应用程序里面的 软件与更新 选择最适合自己的版本，然后直接点击 应用更改，等待一会

安装完毕后，直接重新启动，然后输入nvidia-smi 成功出现以下画面：

3. 安装 cudatookit 12.2

由于显卡最高支持到CUDA 12.2
这个版本以前的CUDA都是兼容的，怕以后再升级麻烦，所以就直接安装支持的最高版本了

CUDA Toolkit Archive的网址为 https://developer.nvidia.com/cuda-toolkit-archive

点击CUDA Toolkit 12.2.0下载相应版本的CUDA12.2.0（选自己想要安装的版本即可，这里以12.2.0为例）

为了少输入点指令，所以就选择了runfile，感觉是最简单的方法了

运行上面指令后，会弹出如下界面，点击Continue，然后再输入accept

在弹出的界面中通过Enter键，取消Driver和535 的安装，然后点击Install，等待

配置CUDA的环境变量
CUDA安装完成后，需要配置变量环境才能正常使用。首先在终端输入

sudo gedit ~/.bashrc

打开.bashrc文件。然后，在.bashrc文件的最后添加以下CUDA环境变量配置信息

export PATH=$PATH:/usr/local/cuda/bin  
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64  
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

最后 ctrl+s 保存

刷新配置

source ~/.bashrc

查看并且测试安装

nvcc -V

查看到CUDA的安装信息，如下图所示，至此CUDA安装成功！

4. 安装cuDNN

cuDNN的网址为 https://developer.nvidia.com/rdp/cudnn-download

选择Ubuntu20.04系统下，CUDA12.X 对应的cuDNN v8.9.6版本，如下图所示：

下载对应版本的 deb 文件，在下载的文件夹里面，使用命令安装

sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.6.50_1.0-1_amd64.deb

安装完成会显示，后面的操作命令，执行对应的命令就行：

sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.6.50/cudnn-local-1998375D-keyring.gpg /usr/share/keyrings/

若没有显示提示命令则执行通用命令：

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

安装对应的依赖包

进入文件夹cd /var/cudnn-local-repo-ubuntu2204-8.9.6.50/，安装对应的依赖包，这些依赖包直接下载安装会失败，都包含在deb文件中，进入上面提到的文件夹安装即可：

sudo dpkg -i libcudnn8_8.9.6.50-1+cuda12.2_amd64.deb
sudo dpkg -i libcudnn8-dev_8.9.6.50-1+cuda12.2_amd64.deb
sudo dpkg -i libcudnn8-samples_8.9.6.50-1+cuda12.2_amd64.deb

验证安装成功

cd /usr/src/cudnn_samples_v8	#  进入文件夹cd /usr/src/cudnn_samples_v8
cp -r /usr/src/cudnn_samples_v8/ $HOME	# 将 cuDNN 示例复制到用户文件
cd $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make	# 编译 cuDNN mnisiCUDNN 示例
./mnistCUDNN	# 运行 mnistCUDNN 示例

返回 Test passed! 则证明安装成功！！！开始深度学习炼丹之旅吧！！！

参考文献

ubuntu20.04安装NVIDIA显卡以及重启黑屏的一种解决方案
Ubuntu 20.04-NVIDIA显卡驱动-安装和卸载-解决黑屏问题
Ubuntu系统安装nvidia（英伟达）驱动后，开机显示器黑屏无法进入桌面
Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程（图文）
Ubuntu20.04配置NVIDIA+CUDA12.2+CUDNN【附所有下载资源】【亲测有效】【非常详细】
Ubuntu 22.04 安装 CUDA 和 cuDNN 及验证