Ubuntu16.04安装CUDA9.0+CUDNN7.0

我是把安装NVIDIA驱动和安装CUDA分开了。在安装CUDA之前我已经安装好了NVIDIA驱动(NVIDIA安装:)

一、安装cuda

1、下载cuda9.0

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal

2、打开终端,进入到cuda9.0目录下(即上一步下载的文件路径下),输入:

$ chmod a+x cuda_9.0.176_384.81_linux.run
$ sudo ./cuda_9.0.176_384.81_linux.run


然后就会进入安装。在安装过程中会询问你一些配置信息,一般直接选择默认选项就行。有一个必须注意的地方是会让你选择是否安装 Graphics Driver for Linux-x86_64:XXXX 版本,这个地方必须选择 no!否则会覆盖之前的驱动。

出现下面界面时,使用Q键跳过

3、安装完毕后,添加环境变量。

$ sudo gedit ~/.bashrc

在打开的文件末尾加上:



export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64

export PATH=$PATH:/usr/local/cuda-9.0/bin

export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0

保存退出

(一般cuda安装路径默认为/usr/local/cuda-9.0)

4、验证安装是否成功

nvcc --version

如果出现下面的内容就说明安装成功了

二、安装cudnn

1、下载cudnnhttps://developer.nvidia.com/rdp/cudnn-archive

注意:下载cudnn需要先注册nvidia账号。

2、下载后,cd进入到文件所在目录,解压:

$ tar -zxvf xxxxxx.tgz

3、解压后会产生一个cuda文件夹,里面有两个文件夹lib64和include,将这两个文件夹复制到/usr/local/cuda-9.0中

 $ sudo cp {include,lib64} /usr/local/cuda-9.0

注意,以上操作是在刚刚解压后的cuda文件夹内。如果没有进入该目录,则需要执行以下命令:

 $ sudo cp /xxx/xxx/cuda/{include,lib64} /usr/local/cuda-9.0

4、执行以下命令


$ cd /usr/local/cuda-9.0/lib64

$ sudo chmod +r libcudnn.so.7.6.1  # 这里要自己去看看自己的.so版本是多少

$ sudo ln -sf libcudnn.so.7.6.1 libcudnn.so.7

$ sudo ln -sf libcudnn.so.7 libcudnn.so

$ sudo ldconfig

5、验证cudnn是否安装成功

cat /usr/local/cuda-9.0/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果出现下面界面,则说明安装成功:

发布了46 篇原创文章 · 获赞 0 · 访问量 5904

猜你喜欢

转载自blog.csdn.net/ly_twt/article/details/104876169