keras转tf并加速（2）ubuntu16.04中tensorflow压缩模型transform_graph编译

Love The Way You Lie 2022-02-03 14:25 171阅读 0赞

1、安装bazel

官网安装指南：[https://docs.bazel.build/versions/master/install-ubuntu.html\#install-with-installer-ubuntu][https_docs.bazel.build_versions_master_install-ubuntu.html_install-with-installer-ubuntu]

发布版本：[https://github.com/bazelbuild/bazel/releases][https_github.com_bazelbuild_bazel_releases]

sudo apt-get install pkg-config zip g++ zlib1g-dev unzip python

wget https://github.com/bazelbuild/bazel/releases/download/0.21.0/bazel-0.21.0-installer-linux-x86\_64.sh

sudo chmod +x bazel-0.21.0-installer-linux-x86\_64.sh

export PATH="$PATH:$HOME/bin"

注意：一定要安装最新版本的bazel，否则在编译代码时可能会报错，具体可见我的另一篇博文：

[https://blog.csdn.net/weixin\_41770169/article/details/90041269][https_blog.csdn.net_weixin_41770169_article_details_90041269]

2、编译代码

下载最新的TensorFlow库  
git clone https://github.com/tensorflow/tensorflow.git  
cd ./tensorflow  
编译graph\_transforms模块，需要等待一段时间，编译很耗费CPU资源和时间  
bazel build tensorflow/tools/graph\_transforms:transform\_graph

参考：[https://blog.csdn.net/shiheyingzhe/article/details/86663889][https_blog.csdn.net_shiheyingzhe_article_details_86663889]

[https_docs.bazel.build_versions_master_install-ubuntu.html_install-with-installer-ubuntu]: https://docs.bazel.build/versions/master/install-ubuntu.html#install-with-installer-ubuntu
[https_github.com_bazelbuild_bazel_releases]: https://github.com/bazelbuild/bazel/releases
[https_blog.csdn.net_weixin_41770169_article_details_90041269]: https://blog.csdn.net/weixin_41770169/article/details/90041269
[https_blog.csdn.net_shiheyingzhe_article_details_86663889]: https://blog.csdn.net/shiheyingzhe/article/details/86663889