查看自己的驱动

$ ubuntu-drivers devices

依赖库的安装

$ sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev
$ sudo apt-get install libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

安装cuda9.0

进入这个地址下载：

然后在网页中往下拉，下载下图中的那个：

image.png

下载后，进入你放这这个文件的文件夹，在此文件夹下操作。

$ sudo dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub

如果以上成功运行的话，会返回OK。

$ sudo apt-get update
$ sudo apt-get install cuda

到这里就应该是安装完了，然后检查一下是否安装成功。

$ cd /usr/local
$ ls

看到下面的文件夹有cuda和cuda-9.0就是我们已经安装成功了。

环境配置

打开终端输入：

$ sudo gedit ~/.bashrc

在打开的配置文件中，将以下内容写入末尾处.

export CUDA_HOME=/usr/local/cuda-9.0
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:$LD_LIBRARY_PATHs
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/lib64"

使环境变量生效.

$ source ~/.bashrc

测试

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

如果结果有GPU的信息，说明安装成功。

安装cudnn7.0

为适应Cudn9.0，这里下载的是CuDnnv7.0.5, for CUDA 9.0（cuDNN v7.0.5 Library for Linux）。
下载地址：https://developer.nvidia.com/rdp/cudnn-archive

解压并复制相应文件

$ tar -xvzf cudnn-9.0-linux-x64-v7.tgz
$ sudo cp cuda/include/cudnn.h   /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

安装后进行测试