我是把安装NVIDIA驱动和安装CUDA分开了。在安装CUDA之前我已经安装好了NVIDIA驱动(NVIDIA安装:)
一、安装cuda
1、下载cuda9.0
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
如果出现下面界面,则说明安装成功: