Tensorflow安装教程

环境

• GTX1080Ti
• Ubuntu14.04LTS（阿里的源）

步骤

1. 安装CUDA8.0
2. 安装Cudnn
3. 安装pip
4. 安装Tensorflow

安装CUDA8.0

总体上的思路参考这篇文章（ubuntu14.04下tensorflow安装指南（GPU）），但是其中坑很多。

• 下载好安装文件，注意，这个安装文件官方有一版本是错误的，现在已经修复。文件名都是cuda_8.0.61_375.26_linux.run，window下是1.4G大小那个是对的。否则直接安装不了。
• 安装时，最好不要选择安装opengl，会和ubuntu冲突不能再次打开图形界面了！！！
• 安装完成后，可能想切换成图形界面，碰到登陆循环的问题，解决方案网上一大堆，但是都没去试。最后我是root下卸载了nvidia的驱动重启就直接成功了。具体原因因为一开始装ubuntu后没有显卡驱动，字都很大模糊，然后先装了驱动。

nvidia-uninstall

坑主要三个，就够受的了。后来问题解决了，找到一篇文章，不错，作者也是用心解决的，不像那些随便抄抄抄的。ubuntu16.04+cuda8.0+cudnn5.1+caffe 安装（基于独立主机&集成显卡&GTX1080Ti）

安装Cudnn

这个好像没什么太多的坑，照着第一个文章走没有什么问题。唯一不确定的是在/usr/local/目录下有两个，一个cuda还有一个cuda-8.0，最后我选择两个都复制了一遍。

安装pip

本身ubuntu就有python2和3,Tensorflow好像基于2的。

sudo apt-get install python-pip
sudo apt-get install python-dev

安装Tensorflow

安装Tensorflow其实很简单，被网上很多的弄得很麻烦，安装网上的一些版本他们都去寻找cuda7.0或者7.5的，所以还是让他自己下载安装吧。对了，网上的那种还要翻墙。

sudo pip install tensorflow-gpu==1.2

测试

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print sess.run(a+b)
42