Ubuntu18.04下安装(1080Ti)NVIDIA驱动、CUDA9.2、cuDNN7.3进行GPU加速

1. 查看相关工具版本的方法

查看NVIDIA驱动版本

nvidia-smi

查看CUDA版本

cat /usr/local/cuda/version.txt
或者 nvcc --version（设置并更新环境变量之后才可用）

查看cuDNN版本

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

信息更新（20190521）

(base) famir@ubuntu:~$ nvidia-smi
Tue May 21 19:29:29 2019
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.56       Driver Version: 418.56       CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 108...  Off  | 00000000:01:00.0  On |                  N/A |
|  0%   42C    P8    19W / 250W |    468MiB / 11178MiB |      0%      Default |
+-------------------------------+----------------------+----------------------++-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1894      G   /usr/lib/xorg/Xorg                            18MiB |
|    0      1960      G   /usr/bin/gnome-shell                          70MiB |
|    0      4407      G   /usr/lib/xorg/Xorg                           245MiB |
|    0      4602      G   /opt/teamviewer/tv_bin/TeamViewer              8MiB |
|    0      6832      G   ...uest-channel-token=16521326733675790024   122MiB |
+-----------------------------------------------------------------------------+
(base) famir@ubuntu:~$ cat /usr/local/cuda/version.txt
CUDA Version 10.0.130
(base) famir@ubuntu:~$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
(base) famir@ubuntu:~$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 5
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)#include "driver_types.h"

2. 安裝或更新NVIDIA驱动

1.安装之前先卸载已经存在的驱动版本：

sudo apt-get purge nvidia*
或者 sudo apt-get remove --purge nvidia*

2.需要禁用 nouveau，只有在禁用掉 nouveau 后才能顺利安装 NVIDIA 显卡驱动，禁用方法就是在 /etc/modprobe.d/blacklist-nouveau.conf 文件中添加一条禁用命令，首先需要打开该文件，通过以下命令打开:

sudo gedit /etc/modprobe.d/blacklist.conf

在最后一行加上: blacklist nouveau ，将Ubuntu自带的显卡驱动加入黑名单。Ctrl +S保存后注意此时还需在终端执行以下命令使禁用 nouveau 真正生效： sudo update-initramfs -u

3.重启電腦后，输入命令sudo service lightdm stop关闭桌面服务，然后就可以安装驱动了。

4.在Ubuntu 18.04上安装NVIDIA有三种方法：

使用标准Ubuntu仓库进行自动化安装
使用PPA仓库进行自动化安装
使用官方的NVIDIA驱动进行手动安装

在這裏使用第一種方法：即使用标准Ubuntu仓库进行自动化安装
这种方法几乎是所有的示例中最简单的方法，也是该教程最为推荐的方法。首先，检测你的NVIDIA显卡型号和推荐的驱动程序的模型。在命令行中输入如下命令：ubuntu-drivers devices，顯示如下：

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001B06sv00001458sd00003752bc03sc00i00
vendor   : NVIDIA Corporation
model    : GP102 [GeForce GTX 1080 Ti]
driver   : nvidia-driver-396 - third-party free recommended
driver   : nvidia-driver-390 - third-party free
driver   : xserver-xorg-video-nouveau - distro free builtin

从输出结果可以看到，目前系统已连接Nvidia GeForce GTX 1080 Ti显卡，建议安装驱动程序是 nvidia-driver-396版本的驱动。如果您同意该建议，请再次使用Ubuntu驱动程序命令来安装所有推荐的驱动程序。
输入以下命令：sudo ubuntu-drivers autoinstall，安装结束，重新启动系统。

注：本次安装过程中，提示老版本驱动有一些包未删除干净，可使用命令sudo apt autoremove删除。

重启后输入以下指令进行验证： nvidia-smi ，若列出了GPU的信息列表则表示驱动安装成功。如下：

Fri Sep 28 11:01:49 2018
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 396.54                 Driver Version: 396.54                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 108...  Off  | 00000000:01:00.0  On |                  N/A |
|  0%   44C    P0    64W / 250W |    309MiB / 11175MiB |      0%      Default |
+-------------------------------+----------------------+----------------------++-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1212      G   /usr/lib/xorg/Xorg                            18MiB |
|    0      1254      G   /usr/bin/gnome-shell                          49MiB |
|    0      1570      G   /usr/lib/xorg/Xorg                           104MiB |
|    0      1749      G   /usr/bin/gnome-shell                          89MiB |
|    0      1796      G   /opt/teamviewer/tv_bin/TeamViewer              2MiB |
|    0      2371      G   /proc/self/exe                                40MiB |
+-----------------------------------------------------------------------------+

或查看设置>详细信息>图形：

注：重装Ubuntu后，使用第一种方法系统推荐396版本:

➜  ~ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001B06sv00001458sd00003752bc03sc00i00
vendor   : NVIDIA Corporation
model    : GP102 [GeForce GTX 1080 Ti]
driver   : nvidia-driver-396 - third-party free recommended
driver   : nvidia-driver-390 - distro non-free
driver   : xserver-xorg-video-nouveau - distro free builtin

于是使用第二种方法自己安装最新的410版本：

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt updateubuntu-drivers devices  #查看自己的显卡及可以安装的驱动版本，显示如下：== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==modalias : pci:v000010DEd00001B06sv00001458sd00003752bc03sc00i00vendor   : NVIDIA Corporationmodel    : GP102 [GeForce GTX 1080 Ti]driver   : nvidia-driver-390 - third-party freedriver   : nvidia-driver-410 - third-party free recommendeddriver   : nvidia-driver-396 - third-party freedriver   : xserver-xorg-video-nouveau - distro free builtinsudo apt install nvidia-driver-410

完成后重启系统。

还可以使用软件和更新来安装或更新驱动：

本部分参考：
Ubuntu 16.04 禁用 nouveau 安装 nvidia显卡驱动
Ubuntu 18.04 NVIDIA驱动安装总结

3. 安装CUDA9.2

安装之前先到NVIDIA官网下载对应版本cuda和cudnn工具文件
https://developer.nvidia.com/cuda-toolkit-archive
https://developer.nvidia.com/cudnn (需注册)
此处下载如下版本文件：

cuda_9.2.148_396.37_linux.run  # cuda文件（1.8G）
cuda_9.2.148.1_linux.run            # cuda补丁（85.6M）
cudnn-9.2-linux-x64-v7.3.1.20.tgz   # cudnn文件（368.1M）

安装依赖：

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev libfreeimage3 libfreeimage-dev
# 否则安装cuda后会出现提示：Missing recommended library: libGLU.soMissing recommended library: libX11.soMissing recommended library: libXi.soMissing recommended library: libXmu.soMissing recommended library: libGL.so

终端输入：sudo sh cuda_9.2.148_396.37_linux.run
空格到底或者输入q退出阅读EULA，除了提问是否安装显卡驱动时选择no，其他均选yes或默认路径。显示如下：

Installing the CUDA Toolkit in /usr/local/cuda-9.2 ...
Installing the CUDA Samples in /home/famir ...
Copying samples to /home/famir/NVIDIA_CUDA-9.2_Samples now...
Finished copying samples.===========
= Summary =
===========Driver:   Not Selected
Toolkit:  Installed in /usr/local/cuda-9.2
Samples:  Installed in /home/famirPlease make sure that-   PATH includes /usr/local/cuda-9.2/bin-   LD_LIBRARY_PATH includes /usr/local/cuda-9.2/lib64, or, add /usr/local/cuda-9.2/lib64 to /etc/ld.so.conf and run ldconfig as rootTo uninstall the CUDA Toolkit, run the uninstall script in /usr/local/cuda-9.2/binPlease see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-9.2/doc/pdf for detailed information on setting up CUDA.***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 384.00 is required for CUDA 9.2 functionality to work.
To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:sudo <CudaInstaller>.run -silent -driverLogfile is /tmp/cuda_install_20157.log

安装补丁：sudo sh cuda_9.2.148.1_linux.run
设置cuda环境变量（比较重要，不然samples跑不起来）：

gedit ~/.bashrc（我的是gedit ~/.shrc,因为我更改了默认shell)
# 在最后加入两行：（按照安装提示修改成自己的路径）export PATH=/usr/local/cuda/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
# 保存退出之后，运行:
source ~/.bashrc（我的是source ~/.zshrc,或者重启终端)

cuda测试：

# 在目录/home/famir/NVIDIA_CUDA-9.2_Samples/1_Utilities/deviceQuery中打开终端
make
./deviceQuery最后出现：Result = PASS 即可。
deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 10.0, CUDA Runtime Version = 9.2, NumDevs = 1
Result = PASS

4. 安装cuDNN7.3

cuDNN 的安装，就是将 cuDNN 包内的文件，拷贝到cuda文件夹中即可。

# 进入文件cudnn-9.2-linux-x64-v7.3.1.20.tgz放置目录，在终端打开，依次执行：
tar -xzvf cudnn-9.2-linux-x64-v7.3.1.20.tgz
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*

部分参考：
NVIDIA官方教程
真实机下 ubuntu 18.04 安装GPU +CUDA+cuDNN 以及其版本选择（亲测非常实用）
Ubuntu18:装GPU TensorFlow