安装tensorflow

1.

gcc降版本

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

sudo apt-get install g++-4.9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
sudo update-alternatives --set cc /usr/bin/gcc
sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo 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 - xzvf cudnn - 8.0 - linux - x64 - v5 . 1.tgz

$ sudo cp cuda / include / cudnn . h / usr / local / cuda / include

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

$ sudo chmod a + 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 ... >>> import tensorflow as tf >>> print(tf.__version__) 0.12 . 0 -rc1