Ubuntu 16.04下配置GPU版CUDA和cuDNN

先介绍一下我自己的配置环境

Ubuntu 16.04

Intel i7 CPU

16G内存

GTX980Ti 显卡

1、安装NVIDIA显卡驱动

（1）先在NVIDIA官网上下载对应的驱动程序，可根据自己的GPU的型号下载相应的.run文件

例如NVIDIA-Linux-x86_64-3xx.xx.run形式的文件名

自己电脑的GPU型号查询可在终端输入：

nvidia-smi

查询。大概在中间位置。也可以输入：

nvidia-settings

（2）禁用开源nouveau驱动（非常重要）

a.如果没有装vim,则先装vim编辑器（当然用gedit也可以，在下面也会介绍到）

安装vim，在终端输入

sudo apt-get update
sudo apt-get install vim

接下来  编辑文件/etc/modprobe.d/disable-nouveau.conf

aa.使用vim（关于vim编辑器的使用，百度上有很多），在终端输入

sudo vim /etc/modprobe.d/disable-nouveau.conf

然后按键盘上的字母 i 进入编辑模式，并在最后添加下面两行

blacklist nouveau
options nouveau modeset = 0

按Esc键退出编辑模式，再输入下面的命令， 回车，保存退出。

:wq

然后输入

sudo ldconfig

再执行下面的命令，使配置生效。

sudo update-initramfs -u

bb.使用gedit，在终端输入：

sudo gedit /etc/modprobe.d/disable-nouveau.conf

然后同样在末尾加上那两句，然后鼠标点save保存，后续步骤则跟用vim一样

验证是否成功禁用nouveau

关掉终端，重新打开，然后输入命令：

 sudo lspci | grep nouveau

如果没有内容显示，则禁用成功。

b.配置kernel以text模式启动

sudo vim /etc/default/grub

编辑文件/etc/default/grub，修改下面两句（操作方法同步骤a，建议用gedit）sudo gedit /etc/default/grub

GRUB_CMDLINE_LINUX_DEFAULT="quiet splash"
GRUB_CMDLINE_LINUX=""

将上面两句修改为：

GRUB_CMDLINE_LINUX_DEFAULT="quietsplash text"
GRUB_CMDLINE_LINUX="rdblacklist=nouveaunouveau.modeset=0"

保存退出，并更新。

sudo update-grub

c.重启电脑，在登录界面直接按 Ctrl+Alt+F1进入控制台

输入用户名和密码登陆后，执行：

sudo service lightdm stop

关闭图形界面（切记，否则英伟达显卡驱动会安装失败），然后cd命令切换到英伟达显卡驱动安装文件NVIDIA-Linux-x86_64-3xx.xx.run所在的目录，执行：

sudo sh NVIDIA-Linux-x86_64-3xx.xx.run

输入以下命令，重启图形界面

sudo service lightdm start

d.结束后再次修改grub文件

将前面修改的第一行改回原来的。第二行可不修改。本人改完是这样的：

GRUB_CMDLINE_LINUX_DEFAULT="quiet splash"
GRUB_CMDLINE_LINUX="rdblacklist=nouveaunouveau.modeset=0"

保存退出后，再次更新

sudo update-grub

2、 安装CUDA以及cuDNN

（1）先在官网上下载对应的CUDA安装文件，建议选择run安装，点这里

自己根据实际情况选择相应的文件。

下载完成后，打开终端切换到安装文件所在的目录，执行：

sudo sh cuda_8.0.61_375.26_linux.run

这里会出现More（0%），意思是你要阅读这些协议，连续按空格键可跳过，按照提示输入accept，第一个安装的时候，是安装图形驱动的，由于前面已经安装了NVIDIA驱动，所以选择no，后面全选yes，建议选择默认安装路径。

（2）环境变量配置

sudo gedit ~./bashrc

将以下内容写到文件尾部。

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

保存后，更新文件

source ~/.bashrc

（3）配置cuDNN

首先在官网上(https://developer.nvidia.com/cuda-downloads)下载CUDA：

下载cuDNN5.1之后切换到下载目录进行解压：

sudo tar -zxvf ./cudnn-8.0-linux-x64-v5.1.tgz 

进入cuDNN5.1解压之后的include目录，在命令行进行如下操作：

cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include  #复制头文件

再将进入lib64目录下的动态文件进行复制和链接：

cd ..
cd lib64
sudo cp lib* /usr/local/cuda/lib64/    #复制动态链接库
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5    #删除原有动态文件
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5  #生成软衔接
sudo ln -s libcudnn.so.5 libcudnn.so      #生成软链接
sudo ldconfig      #使配置生效

（4）测试CUDA的samples

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

如果显示一些关于GPU的信息，怎说明安装成功。

可通过

nvcc --version

查看CUDA版本。