安装显卡驱动

在ubuntu16.04中，更换驱动非常方便，去

系统设置->软件更新->附加驱动->切换到最新的NVIDIA驱动即可。应用更改->重启

选区_052.png-132.8kB

再运行nvidia-smi来看看

选区_053.png-55.8kB

安装CUDA

去https://developer.nvidia.com/cuda-toolkit-archive，根据自己的系统下载对应安装包,下载的是runfile文件，而不是deb文件。下载好以后，运行

./cuda_8.0.44_linux.run

开始安装。安装开始以后，首先是一个协议，一直按回车到底以后，输入accept。其他的操作如下所示

======================

Do you accept the previously read EULA?

accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48?

(y)es/(n)o/(q)uit: n

Install the CUDA 8.0 Toolkit?

(y)es/(n)o/(q)uit: y

Enter Toolkit Location

[ default is /usr/local/cuda-8.0 ]:

Do you want to install a symbolic link at /usr/local/cuda?

(y)es/(n)o/(q)uit: y

Install the CUDA 8.0 Samples?

(y)es/(n)o/(q)uit: y

======================

安装完毕后，再声明一下环境变量，并将其写入到 ~/.bashrc 的尾部:

export CUDA_HOME=/usr/local/cuda

export PATH=$PATH:$CUDA_HOME/bin

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

保存退出，运行 source ~/.bashrc

测试是否安装成功

cd /usr/local/cuda/samples/1_Utilities/deviceQuery

make

./deviceQuery

结果如下

选区_054.png-169.8kB

之前有看到说要把gcc改成4.9的。不过我现在用5.4的也可以编译，就是有几个警告，不知道会不会有什么大的影响。

cuDNN安装

如果要使用gpu来对tensorflow进行加速，除了安装CUDA以外，cuDNN也是必须要安装的。跟cuda一样，去https://developer.nvidia.com/cudnn下载cuDNN的安装包。不过这次没法直接下载，需要先注册，然后还要做个调查问卷什么的，稍微有点麻烦。我下的是cuDNN v5.1 Library for Linux这个版本。不要下cuDNN v5.1 Developer Library for Ubuntu16.04 Power8 (Deb)这个版本,因为是给powe8处理器用的，不是amd64.

下载下来以后，发现是一个tgz的压缩包，使用tar进行解压

tar -xvf cudnn-8.0-linux-x64-v5.1.tgz

安装cuDNN比较简单，解压后把相应的文件拷贝到对应的CUDA目录下即可

cp cuda/include/cudnn.h /usr/local/cuda/include/

cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

chmod a+r /usr/local/cuda/include/cudnn.h

chmod a+r /usr/local/cuda/lib64/libcudnn*

添加环境变量

## cuDNN

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

保存退出，运行source ~/.bashrc

查看当前cudnn的版本