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

时间：2023-12-19 21:56:50

一、安装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..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.           #删除原有动态文件

sudo ln -s libcudnn.so.6.0. libcudnn.so.     #生成软链接

sudo ln -s libcudnn.so. libcudnn.so            #生成软链接

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

6、完毕

相关文章