ubuntu16.04+GTX1080TI+CUDA9.0+CUDNN7.0

@ 环境:hp,ubuntu16.04 ,GTX1080TI

一、安装NVIDIA显卡

打开命令窗口(ctrl+Alt+T

sudo apt-get purge nvidia*
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384 nvidia-settings

安装完毕后重启电脑,运行nvidia-smi,查看生效的显卡驱动
ubuntu16.04+GTX1080TI+CUDA9.0+CUDNN7.0

二、安装CUDA

1、官网下载cuda9.0(cuda_9.0.176_384.81_linux.run):下载链接

2、cd 到cuda下载目录进行安装。

sudo sh cuda_9.0.176_384.81_linux.run

(1)按q键跳过安装须知,输入accept接受。

(2)问你驱动装不装,输入n,选择不装,上面装过了。

(3)问你CUDA装不装,输入y,选择安装。回车键安装在默认位置(也可以自己选择位置安装)。

(4)问你要不要创建链接,输入y,选择创建。

(5)问你sample装不装,因为我们只是调用CUDA,不去写CUDA程序,输入n,选择不安装。

3、添加路径库:

(1)打开bashrc:

sudo vim ~/.bashrc

(2)添加路径:

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

(3)使之生效:

source ~/.bashrc

三、安装CUDNN7.0

1、官网下载CUDNN7.0(cudnn-9.0-linux-x64-v7.4.1.5.tgz):下载链接

2、下载cudnn7.0的linux包(tgz格式)。

3、cd 到下载路径解压文件:

tar -xzvf cudnn-9.0-linux-x64-v7.4.1.5.tgz

4、复制文件:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

5、添加权限:

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

到此,全部安装结束,接下来就可以安装tensorflow,caffe等深度学习框架了。
四、卸载及删除安装

1、cudnn:直接remove掉就可以了。

2、cuda:sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl

3、nvidia驱动:nvidia-uninstall

五、查看版本

1、CUDA :

cat /usr/local/cuda/version.txt

2、 CUDNN :

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

参考:

1、https://blog.csdn.net/weixin_40294256/article/details/79157838

2、https://blog.csdn.net/mtllyb/article/details/79505012