踩了不少坑,现将正确方法记录如下:
安装bezel
安装方法参考官方文档。
值得注意的是,目前貌似是由于bazel的bug,tensorflow的1.4版本编译需要用bazel 0.5.4~0.8.0,再往上的版本会编译失败,具体参考tensorflow的这个issues。
另外,bazel 0.12也有问题,具体参看这个issues。因此在这里选择bazel 0.11.1版本。
安装完成后使用命令查看bazel版本
sudo bazel version
显示
Build label: 0.11.1
Build target: bazel-out/k8-opt/bin/src/main/java/com/google/devtools/build/lib/bazel/BazelServer_deploy.jar
Build time: Tue May 14 07:48:23 +50148 (1520362424903)
Build timestamp: 1520362424903
Build timestamp as int: 1520362424903
表明安装成功。
编译TensorFlow源码
从github上下载tensorflow源码,或者切换到要保存代码的目录使用命令
sudo git clone git@github.com:tensorflow/tensorflow.git
然后切换到tensorflow目录,配置编译选项,然后开始build
./configure
这里注意若要编译tensorflow GPU版需要手下安装CUDA以及cuDNN,这里不在赘述。
配置完成之后,先安装tensorflow for python ,如果已经有了python版本可略去这一步
sudo bazel build --config=mkl <