Ubuntu16.04下安装CUDA和cudnn

首先是安装CUDA

我的系统是Ubuntu16.04, 驱动是NVIDIA GETFORCE 2080 Ti

准备工作

1.1 禁用nouveau
ubuntu 16.04默认安装了第三方开源的驱动程序nouveau，安装nvidia显卡驱动首先需要禁用nouveau，不然会碰到冲突的问题，导致无法安装nvidia显卡驱动。指令如下:
sudo gedit /etc/modprobe.d/blacklist.conf
打开文件，在最后添加如下两行：
blacklist nouveau
options nouveau modeset=0

1.2 更改系统修改
sudo update-initramfs -u

1.3 验证nouveau是否已禁用
lsmod | grep nouveau

下载驱动文件并使用指令安装

2.1 从NVIDIA官网 https://www.geforce.cn/drivers 上下载相对应的驱动
根据自己电脑的显卡型号选择相应的驱动，建议不要选择最新的，往往最新的驱动可能会不适配你当前电脑，文件格式为：NVIDIA-Linux-x86_64-xxx.run

2.2 按下ctrl+alt+f1进行命令行界面，并登陆

2.3 关闭图形界面
sudo service lightdm stop

2.4 卸载原有驱动
sudo apt-get remove nvidia-*

2.5 给安装问价添加权限
sudo chmod a+x NVIDIA-Linux-x86_64-xxx.run

2.6 运行安装文件
sudo ./NVIDIA-Linux-x86_64-xxx.run -no-x-check -no-nouveau-check -no-opengl-files

其中：
-no-x-check：安装驱动时关闭X服务
-no-nouveau-check：安装驱动时禁用nouveau
-no-opengl-files：只安装驱动文件，不安装OpenGL文件

在安装过程中会出现：

1、he distribution-provided pre-install script failed! Are you sure you want to continue? YES

2、Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later? NO

3、Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up. YES

2.7 安装成功，返回图形界面
sudo service lightdm start

检测是否成功
nvidia-smi
若出现一个框，显示CUDA 驱动版本，内存大小，则表示安装成功

安装CUDNN

首先一定要确定和你安装好的CUDA版本一致，可从https://developer.nvidia.com/rdp/cudnn-archive下载，这其中的四个文件都要下载下来，笔者当时以为下一个就行，结果一直出问题，给弄傻了

将下载好的cuDNN Library for Linux (x86_64) 解压，复制解压出来的文件到安装好的CUDA环境中
tar -xvf xxx.tar
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

更改权限
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

安装deb包
sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb
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

检验安装成功与否

方法一
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

方法二

cp -r /usr/src/cudnn_samples_v8/ /home/wdong/
cd /home/wdong/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN

参考
[1]:https://blog.csdn.net/wanzhen4330/article/details/81699769
[2]:https://blog.csdn.net/qq_41207924/article/details/110206687
[3]:https://blog.csdn.net/caicaiatnbu/article/details/87626491