Ubuntu16.04安装Nvidia显卡驱动+Cuda8.0+Cudnn6.0

一、安装Nvidia显卡驱动（gtx1050ti）

参考链接：Ubuntu16.04.2 LTS 64bit系统装机记录中的显卡驱动安装部分。

 

二、安装Cuda8.0

 

1、确定自己的系统信息，以Ubuntu16.04.2 LTS 64bit为例，到官网https://developer.nvidia.com/cuda-downloads下载相应的安装文件

2、下载后在当前目录运行命令安装

sudo ./cuda_8.0.61_375.26_linux.run.26_linux-run
安装过程中基本上选择默认路径以及选Yes，但是提示是否安装Nvidia驱动时一定选择No，因为此前已经安装了相应的显卡驱动

 

3、安装成功后，声明一下环境变量，并将其写入到 ~/.bashrc 的尾部:

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

 

4、设置环境变量

终端输入：sudo gedit /etc/profile
文件末尾加入：export PATH=“/usr/local/cuda/bin:$PATH”
运行：source /etc/profile，查看是否有误

 

5、设置动态链接库

终端输入：sudo gedit /etc/ld.so.conf.d/cuda.conf
添加：/usr/local/cuda/lib64
执行生效：sudo ldconfig

 

6、测试cuda的samples

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
make
sudo ./deviceQuery

 

7、完毕

 

三、安装Cudnn6.0

 

1、确定已经安装的Cuda版本，以Cuda8.0为例，到官网https://developer.nvidia.com/rdp/cudnn-download下载相应的库文件（下载前需要登录网站，因此需要进行注册）

2、下载后进行解压

3、进入include文件夹，执行

sudo cp cudnn.h /usr/local/cuda/include/  #复制头文件

 

4、进入lib64文件夹，执行

sudo cp lib* /usr/local/cuda/lib64/  #复制动态链接库

 

5、执行

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6           #删除原有动态文件
sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6     #生成软链接
sudo ln -s libcudnn.so.6 libcudnn.so            #生成软链接

 

注意！！软链接中的数字6.0.21等参照文件夹lib64中原有的软链接名

6、完毕