bazel编译tensorflow,生成tflite

本文介绍了如何下载tensorflow源码,配置bazel版本,安装依赖,以及详细步骤来编译freeze_graph和tflite。强调了 bazel与tensorflow版本匹配的重要性。

注意:在编译过程中出现的问题基本都可以归结为bazel与tensorflow版本不匹配。

1. 首先下载tensorflow源码。

git clone https://github.com/tensorflow/tensorflow

2.进入下载好的tensorflow目录:

3.打开configure.py,查看该源码对应的bazel版本。非常重要

我这里是2.0.0

4.下载bazel

https://github.com/bazelbuild/bazel/releases/tag/2.0.0

5.安装编译需要的依赖项:

pip install numpy
pip install keras_applications
pip install keras_preprocessing

可能还需要安装zip、gcc.

自此,环境搭建完成。
下面是编译tflite


tflite文件的生成大致分为3步:

    1. 在算法训练的脚本中保存图模型文件(GraphDef)和变量文件(CheckP
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值