Ubuntu22.04安装CUDA、cudnn详细步骤

在Ubuntu系统中，使用nvidia-smi命令可以看到当前GPU信息，在右上角可以看到CUDA Version，意思是最大支持的CUDA版本号。
上一篇文章已经安装了显卡驱动，这次继续安装CUDA、cudnn。

拟安装：
cuda 11.7
cuDNN v8.9.0 (for CUDA 11.x)

安装CUDA

打开官网：https://developer.nvidia.com/cuda-toolkit-archive
选择CUDA 11.7，如下图选择“Lunix”及对应的架构、系统、版本和安装形式：

选择runfile（local）形式，网站会给我们安装的命令：
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
sudo sh cuda_11.7.0_515.43.04_linux.run

依次在终端中输入。

第一条命令：

第二条命令：
注意！由于我们之前以及安装过驱动了，这次不要再安装驱动了。

输入accept

在Driver这里按ENTER勾掉！
然后选择Install

最后，打开终端，配置环境变量：

vim ~/.bashrc

最后添加：

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

输入:wq退出。
打开终端输入：

source ~/.bashrc

完成！

---

最后终端输入nvcc -V即可查看CUDA版本：

安装cudnn

下载安装文件

打开并登陆 https://developer.nvidia.com/rdp/cudnn-download
选择下载 cuDNN v8.9.0 (for CUDA 11.x)，
再选择 Local Installer for Ubuntu22.04 x86_64 (Deb)，
就开始下载了。下载完后，建议把deb文件移动到home主目录。

---

安装

打开上方的 Installation Guide 可查看安装方法。
其中我们已经安装了显卡驱动程序和CUDA。
如果系统还没有安装zlib包，可先sudo apt-get install zlib1g安装一下。

安装步骤：
1.打开deb文件存放目录，右键打开终端，输入(deb文件名和你下载的一致)：
sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.0.131_1.0-1_amd64.deb

2.Import the CUDA GPG key.

输入：
sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.0.131/cudnn-local-2063C34E-keyring.gpg /usr/share/keyrings/

3.刷新库:
sudo apt-get update

4.Install the runtime library.

注意，这里的libcudnn8和cuda版本的配对是指定的，可通过apt-cache policy libcudnn8命令查看。
我这里应该使用libcudnn8=8.9.0.131-1+cuda11.8

输入：
sudo apt-get install libcudnn8=8.9.0.131-1+cuda11.8

5.Install the developer library.

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

6.Install the code samples and the cuDNN library documentation.

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

验证是否安装成功

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

注：如果上面的make命令提示缺少FreeImage.h，运行：sudo apt-get install libfreeimage3 libfreeimage-dev

成功：