ubuntu 16.04安装cuda以及tensorflow-gpu

日期：2018.05.11

1.检查系统是否带有Nvida 显卡

lspci | grep -i nvidia

非空即可
安装驱动：
http://www.nvidia.com/download/driverResults.aspx/132256/en-us
下载合适的deb包后

i) `dpkg -i nvidia-diag-driver-local-repo-ubuntu1604-384.125_1.0-1_amd64.deb’ for Ubuntu 
ii) `apt-get update`
iii) `apt-get install cuda-drivers` 
iv) `reboot`

2.安装Tensorflow-gpu

sudo apt-get install python3-pip
sudo pip3 install tensorflow-gpu

安装成功后，我们会发现无法使用tensorflow，这是因为我们还没有安装cuda以及cudnn.报错如下：

···
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
···

可以看到 该pip安装的tensorflow-gpu依赖cuda9.0

3.安装Cuda9.0

1.下载deb文件：
链接：https://pan.baidu.com/s/1hNysVskSmaQCpbF-cZDVKQ 密码：vxxh
2.把deb文件上传到linux的某目录下
3.

sudo dpkg -i cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-9-0

4.安装cudnn for cuda9.0

安装完cuda,tensorflow还是跑不起来的，报错如下，此时需要安装cudnn 7.

ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory

1.下载deb 文件
链接：https://pan.baidu.com/s/1ndQSxdj_wxXK9a2vK6XQ1Q 密码：ojnn
2.把deb文件上传的linux某目录下
3.

sudo dpkg -i libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb

5.验证

import tensorflow as tf
tf.Session().run(fetch=["Hello tensorflow"])

成功即可。