Ubuntu 16.04 Tensorflow 使用源码安装

如果想要tensorflow可以调用GPU，推荐使用这种方式来安装，前提是显卡的驱动已经成功安装。

1、安装cuda：

网址：https://developer.nvidia.com/cuda-downloads

$ sudo dpkg-icuda-repo-ubuntu1504-7-5-local_7.5-18_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

2、安装cudnn：

网址：https://developer.nvidia.com/cudnn

$ tar xvzf cudnn-7.0-linux-x64-v4.0-prod.tgz
$ sudo cp cuda/include/cudnn.h/usr/local/cuda-9.0/include
$ sudo cp cuda/lib64/libcudnn*/usr/local/cuda-9.0/lib64
$ sudo chmod a+r/usr/local/cuda-9.0/lib64/libcudnn*
$ sudo vim ~/.bashrc
$ exportLD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

（-9.0是CUDA的版本号，安装完CUDA后，/usr/local 下会出现CUDA和CUDA-9.0两个目录。我买的教材中以上几个命令均没有‘-9.0’，导致此错误纠缠了很久）

注意：要搞清楚自己安装的CUDA和cuDNN版本号，第5步会用到。

 

3、获取tensorflow：

下载tensorflow源文件

$ gitclone--recurse-submodules https://github.com/tensorflow/tensorflow

4、安装Bazel：

Bazel是Google开源构建工具，类似于Make的工具，用来编译构建tensorflow

Bazel下载地址：

网址：https://github.com/bazelbuild/bazel/releases

$ sudo apt-get installpkg-config zip g++zlib1g-dev unzip
$ chmod +xbazel-version-installer-os.sh
$ ./bazel-version-installer-os.sh--user
$ exportPATH="$PATH:$HOME/bin"

5、安装tensorflow其他的依赖：

$ sudo apt-get installpython-numpy swig python-dev python-wheel

6、配置tensorflow：

进入tensorflow目录

$ sudo ./configure

PS：在配置过程中，不需要的组件可以选择no，否则会给安装带来极大麻烦。

重点关注：

Do you wish to build TensorFlow with GPU support?  Y  （这里GPU也可能显示为CUDA）

出现Configuration finished即代表配置完成。

 

7、使用Bazel编译构建：

$ bazel build -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-mfpmath=both --copt=-msse4.2 --config=cuda -k //tensorflow/tools/pip_package:build_pip_package

8、生成.whl安装包：

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

 

9、使用pip安装：

$ sudo pip install/tmp/tensorflow_pkg/tensorflow-1.1.0-py2-none-any.whl