Install Tensorflow from Source Code

2016-08-19

Install bazel

Install java jdk

Required Java version:

Java JDK 8 or later (JDK 7 is still supported but deprecated).

In centos we can simply install this by command:

1	yum install -y java-1.8.0-openjdk

set JAVA_HOME environment variables in .bashrc:

1	export JAVA_HOME=/usr/lib/jvm/java-openjdk

Download bazel source code

Download from github and compile:

1
2
3

git clone https://github.com/bazelbuild/bazel.git
cd bazel
./compile.sh

Set bazel to PATH

ADD bazel-bin/src/bazel to .bashrc

In my server as:

1	export PATH=$PATH:/root/bazel/output

Now, we have install the bazel in centos.

If you use Ubuntu OS, you can install it form apt repository. In this blog, I will not tell the detailed, which you can reference the link:

1	http://bazel.io/docs/install.html#compiling-from-source

Install Tensorflow

Download source code

Clone the tensorflow from github:

1	git clone https://github.com/tensorflow/tensorflow

Configure for installation

1 2	cd tensorflow/ ./configure

Then, it will tell you which environment to set. such as:

Please specify the location of python. [Default is /usr/local/anaconda2/bin/python]:
Do you wish to build TensorFlow with Google Cloud Platform support? [y/N] n
No Google Cloud Platform support will be enabled for TensorFlow
Found possible Python library paths:
/usr/local/anaconda2/lib/python2.7/site-packages
Please input the desired Python library path to use. Default is [/usr/local/anaconda2/lib/python2.7/site-packages]
/usr/local/anaconda2/lib/python2.7/site-packages
Do you wish to build TensorFlow with GPU support? [y/N] y
GPU support will be enabled for TensorFlow
Please specify which gcc should be used by nvcc as the host compiler. [Default is /usr/bin/gcc]:
Please specify the Cuda SDK version you want to use, e.g. 7.0. [Leave empty to use system default]:
Please specify the location where CUDA toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:
Please specify the Cudnn version you want to use. [Leave empty to use system default]:
Please specify the location where cuDNN library is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:
Please specify a list of comma-separated Cuda compute capabilities you want to build with.
You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus.
Please note that each additional compute capability significantly increases your build time and binary size.
[Default is: "3.5,5.2"]:
Setting up Cuda include
Setting up Cuda lib64
Setting up Cuda bin
Setting up Cuda nvvm
Setting up CUPTI include
Setting up CUPTI lib64
Configuration finished

Create the pip package and install

When building from source, you will still build a pip package and install that.

# To build with GPU support:
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
pip install /tmp/tensorflow_pkg/tensorflow-0.10.0rc0-py2-none-any.whl

Setting up TensorFlow for Development

If you’re working on TensorFlow itself, it is useful to be able to test your changes in an interactive python shell without having to reinstall TensorFlow.

To set up TensorFlow such that all files are linked (instead of copied) from the system directories, run the following commands inside the TensorFlow root directory:

bazel build -c opt //tensorflow/tools/pip_package:build_pip_package
# To build with GPU support:
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
mkdir _python_build
cd _python_build
ln -s ../bazel-bin/tensorflow/tools/pip_package/build_pip_package.runfiles/org_tensorflow/* .
ln -s ../tensorflow/tools/pip_package/* .

Such as change the version from 0.10.0 to 0.11.0 filetensorflow/tools/pip_package/setup.py:

1	_VERSION = '0.11.0rc0'

Then, do the following:

1	python setup.py develop

Now, we can build our develop tensorflow:

1 2	bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg pip install /tmp/tensorflow_pkg/tensorflow-0.11.0rc0-py2-none-any.whl

Verify the installation

[root@node-4 site-packages]# pip show tensorflow
Metadata-Version: 2.0
Name: tensorflow
Version: 0.11.0rc0
Summary: TensorFlow helps the tensors flow
Home-page: http://tensorflow.org/
Author: Google Inc.
Author-email: opensource@google.com
Installer: pip
License: Apache 2.0
Location: /usr/local/anaconda2/lib/python2.7/site-packages
Requires: numpy, mock, protobuf, wheel, six

We can find the version changed from the 0.10.0tc0 to 0.11.0rc0.

Reference

official link:this.