Ubuntu16 安装gpu模式的tensorflow

1. 准备工作

确认自己的电脑上安装了nividia驱动，以及cuda，cudnn，具体安装方法在另一片文章中有介绍

2. pip安装tensorflow

ubuntu16自带的Python2.7 和Python3.4，最好使用python3,运行

pip3 install tensorflow-gpu

可能的坑

ImportError:libcudnn.so.6.0:cannot open shared object file:No such file or directory
打开/usr/local/cuda/lib64,发现现存的文件为libcudnn.so.5.0
tensorflow-gpu最新版本是1.3，需要cuda8以及cudnnv6的支持，而笔者的电脑上安装的为cuda8和cudnnv5，如果不想更新cudnn，就需要安装gensorflow-gpu1.2版本

pip3 uninstall tensorflow-gpu
pip3 install tensorflow-gpu==1.2

再次打开python3，输入import tensorflow，就不会报错了

3. 最后提醒

不要使用源码安装！不要使用源码安装！不要使用源码安装！
笔者使用源码安装了三遍，折腾了一整晚，中间各种坑，最后也没安装成功。最好使用pip安装，一条命令搞定。是有多跟自己过不去非得用源码安装。

4. 更新

首先，感谢天瑞老哥，用了整整一中午帮我解决问题。
这一次，尝试在anaconda_python2.7环境中配置tensorflow.

1)安装anaconda

在conda官网下载对应python2.7版本的安装包,或前往镜像网站下载对应版本
[]!https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
python2.7对应Anaconda2-4.3.1

bash 安装包路径

添加路径

echo 'export PATH="~/anaconda2/bin:$PATH"' >> ~/.bashrc
source ~/.bashrc

2)conda环境管理

创建python2.7的运行环境，这里默认安装python2.7，如安装python3.4则需加上python==3.4

conda create --name tensorflow_py27

安装成功后，激活conda环境

source activate tensorflow_py27

3)安装tensorflow

pip install tensorflow_gpu

安装完成进入python后，发现问题来了：

File "/home/*/.local/lib/python2.7/site-packages/tensorflow/python/client/client_lib.py", line 35...
ImportError: libcudart.so.7.0: cannot open shared object file: No such file or directory

之后，无论笔者安装哪个tensorflow版本，都会报这个错误。
原因出在.local上，发现此时并没有在conda文件夹下运行，这说明之前在本地可能安装过另一个版本的tensorflow。在home目录下执行

pip uninstall tensorflow

原来n年前安装过tensorflow0.5。。。
好啦，之后再在conda下安装tensorflow就没问题了