TensorFlow源码安装

前言

TensorFlow如果能二进制包安装，我真的不想选择自己编译，但是情况不由人，好不容易找到一台服务器，CPU不支持AVX指令集，安装的release版本运行到import tensorflow as tf 就会报指令异常。网上建议源码安装才能解决，于是无奈，想着应该是非常繁琐的事情，但事实上除了耗时外，其他都还好，这里记录一下，以备后用。

编译安装过程

安装版本构建工具 bazel

bazel是google的版本构建工具，网上褒贬不一，直接下载安装包进行安装：

wget https://github.com/bazelbuild/bazel/releases/download/0.13.0/bazel-0.13.0-installer-linux-x86_64.sh
chmod +x ./bazel-0.13.0-installer-linux-x86_64.sh 
./bazel-0.13.0-installer-linux-x86_64.sh 

这个步骤没有碰到问题。

安装工具包

sudo apt install python-numpy swig python-dev python-wheel

numpy是python的科学计算库，swig是C库外部语言封装调用接口库，后面几个应该与python包管理相关，不细究。

下载TensorFlow源码

使用git来克隆TensorFlow代码仓库：

git clone --recurse-submodules https://github.com/tensorflow/tensorflow

比较耗时的操作，文件比较多。

配置TensorFlow

进入tensorflow目录，然后执行./configure，开始配置，除了jemalloc选项外，我都选了N

编译TensorFlow

执行以下命令，启动编译，过程非常慢，要耐心等待：

bazel build -c opt //tensorflow/tools/pip_package:build_pip_package

创建PIP安装包

完成编译后，将tensowflow制作成pip安装包：

bazel-bin/tensorflow/tools/pip_package/build_pip_package ~/

安装TensorFlow包

最后，通过pip命令来安装前一步生成的安装包，如果没有先安装python-pip：

pip install ~/tensorflow-1.8.0-cp27-cp27mu-linux_x86_64.whl 

增加编译选型以支持额外CPU指令

自己编译的版本可以正常使用，但运行时出现了如下的命令行提示：

原来这款CPU虽然不支持AVX，但是支持SSE，于是决定重新编译，在编译时增加了额外的编译选项，如下：

build -c opt --copt=-msse4.1 --copt=-msse4.2 //tensorflow/tools/pip_package:build_pip_package

重新打包，安装。

暂时结束

至此，整个源码安装TensorFlow过程全部结束，暂无发现问题，待后续有问题时再补充。