Ubuntu20.04进行CUDA11.0及对应CUDNN安装

Ubuntu20.04进行CUDA11.0及对应CUDNN安装：

基本步骤：

1.安装nvidia显卡驱动

可直接通过：软件和更新->附加驱动
选择满足CUDA版本的nvidia专有驱动->应用更改，重启即可。
（CUDA版本和显卡驱动版本对应关系：）

2.安装CUDA

2.1 CUDA下载

通过NVIDIA官网,根据硬件/系统条件，选择runfile(local)，根据提供的网址下载.run文件。

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
sudo sh cuda_11.0.2_450.51.05_linux.run

PS：
官方下载方式为通过终端运行wget下载。但笔者建议复制网址，通过网页下载，笔者通过wget下载时，到最后几秒出现段错误（核心已转储）问题，终止下载，重新运行会重新从头下载。

2.2 CUDA安装

下载完成后执行以下语句：

sudo sh cuda_11.0.2_450.51.05_linux.run

选择continue，输入accept，根据需要选择是否需要安装相应nvidia驱动（PS:在第1步安装了驱动则建议在这里取消（空格）driver安装，不然可能安装失败。）其他默认（均选中）即可。
选择install回车。
安装完成后：
通过gedit ~/.bashrc或vim ~/.bashrc等，打开~/.bashrc文件，在最后添加：

export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

保存关闭，并source ~/.bashrc。

3.安装CUDNN

参考NVIDIA cuDNN v8 deb方法安装和卸载教程(Linux/Ubuntu)

3.1 CUDNN下载

到官网找到需要的cudnn版本，下载cuDNN Runtime xxx/ cuDNN Developer xxx/ cuDNN Code xxx三个.deb文件（下载后为libcudnn8_8xxx.deb / libcudnn8-debxxxx.deb / libcudnn8-samplesxxx.deb）。

3.2 CUDNN安装

依次执行以下语句进行安装：

sudo dpkg -i llibcudnn8_8.0.5.39-1+cuda11.0_amd64deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.0_amd64.deb

3.3 CUDNN测试

通过以下语句进行测试：

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

出现Test passe则表明cuDNN已通过程序测试，完成安装！