安装tensorflow

1.

gcc降版本

ubuntu的gcc编译器是5.4.0，然而cuda8.0不支持5.0以上的编译器，因此需要降级，把编译器版本降到4.9：
在terminal中执行：

sudo apt-get install g++-4.9sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30sudo update-alternatives --set cc /usr/bin/gccsudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30sudo update-alternatives --set c++ /usr/bin/g++

2. 安装Cuda 8.0

下载并运行cuda_8.0.61_375.26_linux.run

3. 安装 CudNN 5.1 for Cuda 8.0

$sudo tar-xzvfcudnn-8.0-linux-x64-v5.1.tgz

$sudocpcuda/include/cudnn.h/usr/local/cuda/include

$sudo cpcuda/lib64/libcudnn*/usr/local/cuda/lib64

$sudochmoda+r/usr/local/cuda/include/cudnn.h/usr/local/cuda/lib64/libcudnn*

$sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

4. 修改环境变量

在终端中输入：gedit ~/.bashrc

打开文件后在文件的最后一行加入如下语句：

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"export CUDA_HOME=/usr/local/cuda

5. 安装Bazel

Bazel是Google开源的一套自动化构建工具，可以通过源的方式安装，也可以通过编译源码安装，这里只介绍最简单的通过源的安装方式。

先安装JDK8

1

2

3

$ sudo add-apt-repository ppa:webupd8team/java

$ sudo apt-get update

$ sudo apt-get install oracle-java8-installer

添加Bazel源的地址

1

2

$ 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

1

2

$ sudo apt-get update

$ sudo apt-get install -y bazel

6. 下载并编译tensorflow，如果是cpu支持版本可以直接下载编译好whl文件

#在过程中会进行一些参数路径选择，根据需要自己配置

cd tensorflow

./configure

#添加--config=cuda表示GPU

bazel build -c opt --config=cuda --define=use_fast_cpp_protos=true//tensorflow/tools/pip_package:build_pip_package

如果安装tensorflow提示未编译使用SSE3、SSE4.1、SSE4.2、AVX、AVX2、FMA的问题 ，则采用下面命令编译

bazel build -c opt --copt=-msse3 --copt=-msse4.1 --copt=-msse4.2 --copt=-mavx --copt=-mavx2 --copt=-mfma //tensorflow/tools/pip_package:build_pip_package



#编译完的whl文件会放到/tmp/tensorflow_pkg目录

bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

#安装编译好的TensorFlow

sudo pip install --upgrade /tmp/tensorflow_pkg/tensorflow-0.12*.whl




好，如果顺利的话，到这里我们已经成功安装好了TensorFlow，那么让我简单测试一下安装是否成功。

1

2

3

4

5

$ python

...

>>> importtensorflow as tf

>>> print(tf.__version__)

0.12.0-rc1