Ubuntu16.04安装cuda及cudnn

时间:2022-08-01 17:21:16
注:本人因工作需要,要在linux系统下搭建深度学习环境ssd,需要安装cudacudnn使用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目录下有includelib64个文件夹。

复制头文件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