Ubuntu16.04安装cuda及cudnn

注：本人因工作需要，要在linux系统下搭建深度学习环境ssd，需要安装cuda和cudnn使用GPU来加速。网上安装cuda方法有不少，实际安装过程中碰到很多问题。我经过多次的安装试验，查找资料，发现一种比较简单的方法，分享给大家，希望对大家能有所帮助。话不多说，开始吧！

一.安装NVIDIA显卡驱动：

安装cuda的时候也会有一个安装NVIDIA驱动的选项，但是如果用cuda里的NVIDIA驱动很容易出问题，因此可以先安装NVIDIA驱动在安装cuda。

1.进入ubuntu16.04的系统设置---软件和更新—附加驱动里面选择使用NVIDIA驱动，如下图：

选择后点击应用，系统就自动安装了NVIDIA驱动，当然首先是要联网，是不是感觉很方便啊。

2．验证NVIDIA驱动是否安装成功：

直接ctrl+alt+T打开终端，在终端里输入nvidia-smi(注意是没有空格的)，我们可以直接看到如下所示，就说明我们的nvidia驱动已经安装成功了。

二.安装cuda8.0

我们下载好cuda后直接放在home目录下，直接使用：

sudosh cuda_8.0.27_linux.run

之后一直按空格到100%后输入accept接受条款，

输入no不安装NVIDIA驱动，因为我们已经在上面安装过了；

输入y安装cuda8.0工具，回车使用默认路径安装cuda8.0工具，

输入y使用’sudo’命令，接着输入密码；

输入n不安装指向/usr/local/cuda的符号连接（也可选择y安装符号连接）；

输入y安装cuda8.0的实例，我们可以用此来检验cuda是否安装成功，回车选择默认路径安装；

接着我们就等待cuda安装成功。

 

三.安装cudnn v5.1

说是安装cudnn但其实就是设置cudnn的路径。我们下载cudnn后放在home目录下，可以使用命令

tar zxvfcudnn-8.0-linux-x64-v5.1.tgz

来解压，我们也可以直接右键解压，解压后cuda目录下有include和lib64两个文件夹。

复制头文件cudnn.h到/usr/local/cuda-8.0/include/目录下：

sudo cp cuda/include/cudnn.h/usr/local/cuda-8.0/include/

复制库文件lib*到/usr/local/cuda-8.0/lib64/目录下：

sudo cp cuda/lib/lib*/usr/local/cuda-8.0/lib64/

当然我们也可以手动复制到相应的目录下。

如果发现文件上有一个小锁，我们可以使用

sudo chmod 777目标文件（赋予文件夹及子文件夹777权限）命令去掉锁后再复制。

 

四.建立软连接；

设置环境变量，终端输入  
sudo gedit /etc/profile (一般使用gedit来修改，很方便，如果你喜欢vim也可以用vim)
在末尾加入
PATH=/usr/local/cuda-8.0/bin:$PATH
export PATH
保存后，创建链接文件
sudo gedit /etc/ld.so.conf.d/cuda.conf
增加下面一行
/usr/local/cuda-8.0/lib64
最后在终端输入sudo ldconfig使链接生效。

五.cuda Samples测试

打开CUDA 8.0 Samples默认安装路径，终端输入
cd /home/its/NVIDIA_CUDA-8.0_Samples （its是我的用户名）
sudo make all –j16 （16核，根据你的电脑配置，核越多运行越快）
出现“unsupported GNU version! gcc versions later than 5.3 arenot supported!”的错误，这是由于GCC版本过高，在终端输入
cd /usr/local/cuda-8.0/include
sudo gedit host_config.h
ctrl+f寻找有“5.3”的地方，只有一处，如下
# if __GNUC__ > 5 || (__GNUC__ == 5 && __GNUC_MINOR__ > 3)
#error -- unsupported GNU version! gcc versions later than 5.3 are notsupported!
将3改成9，这样在GCC版本大于5.9的时候才报错，即
# if __GNUC__ > 5 || (__GNUC__ == 5 && __GNUC_MINOR__ > 9)
保存退出，继续在终端输入
sudo make all –j16 （16核）
完成后继续向终端输入
cd bin/x86_64/linux/release
./deviceQuery
完成之后结果如下图所示，表示成功安装cuda。