Ubuntu16.04安装Tensorflow（GPU）简明教程

安装Cuda+cuDNN可参考我上一篇博客：http://blog.csdn.net/Briliantly/article/details/79560477

安装tensorflow的方式有很多种，这里介绍的是通过Bazel编译tensorflow源码的方式来安装tf。
ps：我使用的版本是r1.7

1.安装Bazel
按照官方教程提供方法安装即可。
链接：https://docs.bazel.build/versions/master/install-ubuntu.html

（1）安装JDK8
添加PPA源

sudo add-apt-repository ppa:webupd8team/java
sudo apt-get update

sudo apt-get install openjdk-8-jdk

安装完成后键入

java -version

若看到Java版本信息则表明JDK安装成功

(2)安装依赖包

sudo apt-get install pkg-config zip g++ zlib1g-dev unzip python3

(3)安装Bazel

• 方法一：

通过Bazel在github上的主页可以找到其安装包。
链接：https://github.com/bazelbuild/bazel/releases

由于前面我们已经安装了JDK8，所以这里选择bazel-<version>-without-jdk-installer-linux-x86_64.sh
其中<version>为Bazel的版本号，我使用版本是的是bazel-0.11.1-without-jdk-installer-linux-x86_64.sh
然后赋予其可执行权限，运行：

sudo chmod +x bazel-0.11.1-without-jdk-installer-linux-x86_64.sh
./bazel-0.11.1-without-jdk-installer-linux-x86_64.sh --user

最后然后添加其路径

export PATH="$PATH:$HOME/bin"

重启一下使其生效

• 方法二

添加Bazel的PPA源，但需要科学上网。

echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -

sudo apt-get update
sudo apt-get upgrade bazel

2.安装Tensorflow
PS：我使用的是Python3.5，所以下面就以python3为例来说明
官方教程链接：https://www.tensorflow.org/get_started/os_setup.html

(1)安装一些依赖项

sudo apt-get install python3-pip python3-dev 
sudo apt install python3-numpy swig python3-dev python3-wheel

(2)下载Tensorflow
Github上的主页：https://github.com/tensorflow/tensorflow
branch处可以选择版本
解压源码，进入解压好的文件夹中，在当前目录下打开终端

./configure

配置安装信息

#这里我用的是python3，python2的话默认即可
Please specify the location of python. [Default is /usr/bin/python]: /usr/bin/python3
Please input the desired Python library path to use.  Default is [/usr/local/lib/python3.5/dist-packages]
#除了CUDA以外，其他平台都选n
Do you wish to build TensorFlow with CUDA support? [y/N]: y
#根据CUDA的版本选择
Please specify the CUDA SDK version you want to use, e.g. 7.0. [Leave empty to default to CUDA 9.0]: 9.1
#根据cuDNN的版本选择
Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 7.0]: 7.1
#使用GCC编译器，这里选n
Do you want to use clang as CUDA compiler? [y/N]: n
Do you wish to build TensorFlow with MPI support? [y/N]: n
Would you like to interactively configure ./WORKSPACE for Android builds? [y/N]: n
#这里略去了一些选项

配置完成后会看到最后出现

Configuration finished

（3）通过pip安装

bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg  

cd ///tmp/tensorflow_pkg
ls

通过ls命令确认安装包名字，然后再通过pip安装，<version>和安装环境有关

sudo pip3 install /temp/tensorflow_pkg/tensorflow-<version>-none-any.whl

(4)测试
进入python，键入以下代码

import tensorflow as tf
hello = tf.constant('hello tensorflow')
sess = tf.Session()
print(sess.run(hello))

若看到如下输出，则表明Tensorflow安装成功

hello tensorflow

愉快的一天又开始了～～～