ubuntu 16.04 lts 安装nVidia驱动和Cuda+Cudnn深度学习环境

nvidia显卡驱动

驱动的话提供离线和在线两种方式，有条件的话在线更省事，

写在最前

必做一步：sudo vim /etc/modprobe.d/blacklist.conf，在文件最后加入：blacklist nouveau，然后执行

 sudo update-initramfs -u
 sudo reboot

这一步主要是禁用nouveau驱动，否则后面装完nvidia驱动之后，执行nvidia-smi会提示：

NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver

如果系统是Desktop版本

在安装之前，快捷键ctrl+alt+f1进入命令行界面，然后输入sudo service lightdm stop关闭桌面。

在线安装

如果要使用Cuda8，则采用375以上版本驱动：
sudo apt-get install nvidia-375
如果要使用Cuda9，则必须采用384以上的版本：
sudo apt-get install nvidia-384

离线安装

在官网下载对于的驱动安装文件，记得选Linux版本，根据自己需要的CUDA版本进行对应的版本选型。

其实只要记住Cuda9必须384以上版本就可以

下载后修改对于的run文件权限：chmod +x NVIDIA.run
用以下命令进行驱动安装：

sudo ./NVIDIA.run -no-x-check -no-nouveau-check -no-opengl-files

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

安装过程中，可能会提示一些“pre-install failed”和“32-bit”之类的信息，无视，继续即可。

检查安装

输入命令：nvidia-smi，如果显示如下结果，这表明显卡驱动安装成功：

rszj-ai@rszj-ai:~$ nvidia-smi 
Sat Jul 21 17:16:41 2018       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 375.66                 Driver Version: 375.66                    |
|-------------------------------+----------------------+----------------------+
| 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  | 0000:01:00.0     Off |                  N/A |
|  0%   40C    P8    21W / 300W |      0MiB / 11171MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
|   1  GeForce GTX 108...  Off  | 0000:02:00.0     Off |                  N/A |
|  0%   32C    P8    13W / 300W |      0MiB / 11172MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID  Type  Process name                               Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

注意事项

这边有一份采坑总结写得很不错，分享给各位：地址

CUDA安装

根据需要的版本下载：官网地址
一般选择runfile(local)，如下图：

对下载的文件添加可执行权限，chmod +x cuda.run
安装：./cuda.run，提示安装nvidia driver选择no即可，因为上一步已经安装了显卡驱动，其他的采用默认选项即可。

Cudnn安装

根据版本自行下载，下载地址：官网链接，需要注册账号，请自行注册和下载。
下载后一般是一个压缩包，类似于cudnn-8.0-linux-x64-v5.1.tgz，然后执行如下命令即可。

tar xvf cudnn-8.0-linux-x64-v5.1.tgz
cp -rdf cuda/* /usr/local/cuda/