参考文档
- https://tensorflow.google.cn/install/install_sources Installing TensorFlow from Sources
- https://tensorflow.google.cn/install/install_linux Installing TensorFlow on Ubuntu
Tensorflow训练环境安装
训练环境概述：
- CPU：Intel i7-4790K
- GPU：1 x Nvidia GTX1060 & 1 x Nvidia GTX1080ti
- 操作系统：Ubuntu 16.04.3 64位
- GPU驱动版本：384.98
- CUDA版本：9.0.176
- cuDNN版本：7.0.4.31
- 环境搭建时间：2018-04-12
源代码编译
下载源代码
git clone https://github.com/tensorflow/tensorflow
cd tensorlfow
git checkout r1.7可以通过git的checkout选项切换到指定的Release
或者直接从github上下载稳定版本，并解压得到源代码。
安装Bazel
参考Bazel官方安装文档的建议，推荐使用apt方式安装Bazel。
安装jdk8
对于Ubuntu16.04：
sudo apt-get install openjdk-8-jdk对于Ubuntu14.04则需要先添加PPA源再进行安装：
sudo add-apt-repository ppa:webupd8team/java
sudo apt-get update && sudo apt-get install oracle-java8-installer添加Bazel安装源
sudo apt-get install curl
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 -安装Bazel
sudo apt-get update
sudo apt-get install bazel如果安装过程中出现连接googleapi网站失败的问题，可以尝试多试几次。
升级Bazel
建议安装完Bazel之后进行版本升级
sudo apt-get upgrade bazel安装Python依赖
sudo apt-get install python-numpy python-dev python-pip python-wheel配置Tensorflow
cd tensorflow-1.7.0
./configure编译pip包
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package手动创建pip包
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg安装pip包
sudo pip install /tmp/tensorflow_pkg/tensorflow-1.7.0-cp27-cp27mu-linux_x86_64.whl验证安装过程是否正确
进入python命令行执行环境
python 执行以下python代码，如果执行正确则表示tensorflow已经正确安装：
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))