红联Linux门户
Linux帮助

Ubuntu16.04安装cuda及cudnn

发布时间:2017-07-26 09:37:23来源:linux网站作者:cdwxx1234
注:本人因工作需要,要在linux系统下搭建深度学习环境ssd,需要安装cuda和cudnn使用GPU来加速。网上安装cuda方法有不少,实际安装过程中碰到很多问题。我经过多次的安装试验,查找资料,发现一种比较简单的方法,分享给大家,希望对大家能有所帮助。话不多说,开始吧!
 
一.安装NVIDIA显卡驱动:
安装cuda的时候也会有一个安装NVIDIA驱动的选项,但是如果用cuda里的NVIDIA驱动很容易出问题,因此可以先安装NVIDIA驱动在安装cuda。
1.进入ubuntu16.04的系统设置---软件和更新—附加驱动里面选择使用NVIDIA驱动,如下图:
Ubuntu16.04安装cuda及cudnn
选择后点击应用,系统就自动安装了NVIDIA驱动,当然首先是要联网,是不是感觉很方便啊。
2.验证NVIDIA驱动是否安装成功:
直接ctrl+alt+T打开终端,在终端里输入nvidia-smi(注意是没有空格的),我们可以直接看到如下所示,就说明我们的nvidia驱动已经安装成功了。
Ubuntu16.04安装cuda及cudnn
 
二.安装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。
Ubuntu16.04安装cuda及cudnn
 
本文永久更新地址:http://www.linuxdiyf.com/linux/32118.html