ubuntu16.04+tensorflow1.4+Cuda8.0+CUDNN7.0 系统安装

踩了不少的坑，花了将近一周才把东西弄出来，分享下踩坑经验，愿来者不再入坑。

参考资料_驱动以及cuda安装:

http://blog.csdn.net/u010837794/article/details/63251725 ;

参考资料_cudnn以及tensorflow安装：http://blog.csdn.net/u014595019/article/details/53732015；

1. 系统环境：

ubuntu16.04 + cuda8.0 + cudnn6.0 + tensorflow1.4 + 1080Ti

cuda 9.0对应的cudnn是7.0，但是tensorflow必须是1.6或者1.5的

cuda 8.0对应的cudnn是6.0，tensorflow必须是1.4

否则会报错。

2. 驱动安装

对于1080Ti的英伟达驱动，需要的驱动程序版本是390，这边必须采用在线安装的方式最为靠谱。

安装需要依次执行以下四句:

sudo add-apt-repository ppa:graphics-drivers /ppasudo apt-get updatesudo apt-get install nvidia- 390 sudo apt-get install mesa-common-dev sudo apt-get install freeglut3-dev

执行完毕之后重启即可.

sudo reboot

驱动检验,执行以函数:

nvidia-settings

或者：

nvidia-smi

如果有NVIDIA X Server Settings弹出，则安装成功

3.  安装Cuda

下载官方cuda8.0的runfile文件即可

在下载目录执行如下语句:

sudo sh cuda_8.0.27_linux.run

需要注意的是，不要安装驱动，驱动之前已经安装过了

安装完毕之后需要添加环境变量：

sudo gedit /etc/profile

打开“profile”文件，在末尾处添加（注意不要有空格，不然会报错):

export PATH = /usr/local/cuda-8.0/bin :$PATH export LD_LIBRARY_PATH = /usr/local/cuda-8.0/lib 64$LD_LIBRARY_PATH

重启电脑：

sudo reboot

测试是否安装成功：

cd /usr/local/cuda- 8.0 /samples/ 1 _Utilities/deviceQuerysudo make. /deviceQuery

如果出现关于GPU的信息，则表示安装成功。

4.  安装CuDNN

CUDNN的版本一定要对。tensorflow1.4对应的版本需要CUDNN6.0.

先去官网下载相应的cuDNN安装包，对其进行解压：

· tar -xvf cudnn-8.0-linux-x64-v5.1.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.hsudo chmod a+r /usr/local/cuda/lib64/libcudnn*

5.  安装tensorflow

首先要卸载原先的tensorflow，

sudo pip uninstall tensorflow

在线安装新的tensorflow：

pip3 install tensorflow-gpu==1.4

安装完成之后测试使用:

import tensorflow as tf

sess = tf.Session()

即可查看是否能够正常使用