Ubuntu18.04+CUDA10.2+cuDNN10.2

Ubuntu18.04+CUDA10.2+cuDNN10.2

显卡驱动

查看电脑驱动情况

$ ubuntu-drivers devices

安装驱动

$ sudo ubuntu-drivers autoinstall

查看驱动版本

$ nvidia-smi

查看Nouveau是否禁用

$ lsmod | grep nouveau

禁用nouveau

$ sodu chmod 666 /etc/modprobe.d/blacklist.conf
$ vim /etc/modprobe.d/blacklist.conf

在最后面加上代码块，保存。

blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off

然后在内核中禁用nouveau

$ echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf
$ sudo chmod 644 /etc/modprobe.d/blacklist.conf
$ update-initramfs -u

重启。

CUDA

安装前准备

查看显卡CUDA支持情况：

https://developer.nvidia.com/cuda-gpus

查看本机显卡

$ lspci | grep -i nvidia

安装CUDA

根据驱动版本选择CUDA版本：
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
CUDA10.2下载：
https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=deblocal
CUDA安装命令：

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
$ sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get -y install cuda

安装后

添加环境变量

$ sudo vim ~/.bashrc

在最后添加

$ export PATH=/usr/local/cuda-10.2/bin:/usr/local/cuda-10.2/NsightCompute-2019.1${PATH:+:${PATH}}
$ export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

重新加载

$ source ~/.bashrc

查看CUDA是否成功安装

$ nvcc -V

或者，输入以下命令验证

$ cd /usr/local/cuda-10.2/samples/1_Utilities/deviceQuery
$ sudo make
$ ./deviceQuery

结果的最后几行为

安装cuDNN

注册并下载以下3个文件
cuDNN Runtime Library for Ubuntu18.04 (Deb)
cuDNN Developer Library for Ubuntu18.04 (Deb)
cuDNN Code Samples and User Guide for Ubuntu18.04 (Deb)

** 要下载deb文件，不要下载tgz文件！（亲测tgz易错）
网址
https://developer.nvidia.com/rdp/cudnn-download
进入到下载文件所在目录

$ sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb

检测cuDNN是否安装成功：
运行/usr/src/cudnn_samples_v7 中的mnistCUDNN sample

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

如果出现 Test passed! 则说明安装成功，结果如图
或者，输入以下命令来检查：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果显示如下图则说明成功