【基于Ubuntu16.04】Nvidia显卡驱动+Cuda8.0+Cudnn6.0

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

驱动包准备：

     【 下载NVIDIA 】https://developer.nvidia.com/accelerated-computing-toolkit

     【 下载CUDA 】  https://developer.nvidia.com/cuda-downloads

     【下载CUDNN】  https://developer.nvidia.com/search/site/cudnn

    【自动匹配NVIDIA驱动，自动下载安装】http://elrepo.org/tiki/nvidia-detect

    【官网推荐CUDA安装教程·英文版】https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#ubuntu-installation

一、安装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、完毕

版权声明：本文为博主原创文章，未经博主允许不得转载。

分类: 计算机视觉

标签: nvidia, cuda, cudnn