如何在Ubuntu上安装CUDA和CUDNN

source:https://www.bilibili.com/video/BV1bW4y197Mo/?spm_id_from=333.999.0.0&vd_source=cb51e6e50c82c7f849cf3713e202e16c

选择CUDA版本

根据pytorch选择CUDA版本

根据网页显示选择CUDA11.8

登录官网

选择版本：

输入代码进行安装：

在终端输入代码进行安装

gedit ~/.bashrc

将下面的代码粘贴到代码的最下方

export PATH=/usr/local/cuda-11.8/bin:${PATH}

export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:${LD_LIBRARY_PATH}

输入代码进行验证是否成功：

source ~/.bashrc 
nvcc --version

显示结果为：

验证成功

CUDNN

登录CUDNN官网，第一次登陆需要注册账号进行登录
按照自己的Linux型号进行选择(在设置-关于中查看)：

下载后在该文件的终端上输入以下代码：

sudo apt-get install zlib1g

sudo dpkg -i cudnn-local-repo-${distro}-8.x.x.x_1.0-1_amd64.deb

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

sudo apt-get update

sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y

sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y

sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y

如何验证cudnn？

cudnn验证：

1. cp -r /usr/src/cudnn_samples_v8/ $HOME

2. cd $HOME/cudnn_samples_v8/mnistCUDNN

3. make clean && make

4. /mnistCUDNN

在第三步出现了这个问题：

解决方式：
sudo apt-get install libfreeimage3 libfreeimage-dev

如果出现Test passed则说明通过了