下载bazel并安装
chmod +x bazel-version-installer-os.sh
/bazel-version-installer-os.sh --user
下载tensorflow源码
git clone --recurse-submodules https://github.com/tensorflow/tensorflow.git
编译tensorflow
进入tensorflow目录并configure, 按照提示选择即可
cd tensorflow
./configure
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
注意configure之前应当检查LD_LIBRARY_PATH等环境变量,不正确的环境变量可能会导致错误
制作whl包并安装
生成python安装包并安装。如果系统中已安装其他版本的tensorflow,此步骤会覆盖系统版本。如果仅编译Android版本,则此步骤可省略
mkdir /tmp/tensorflow_pkg
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
python -m pip install /tmp/tensorflow_pkg/tensorflow-0.10.0-py2