如何在build.gradle中声明特定版本的tensorflow-lite nightly依赖?
解决TensorFlow Lite Nightly版本依赖匹配问题
我之前也碰到过一模一样的版本不兼容坑——训练用的tf-nightly版本和Android端TFLite依赖版本对不上,导致推理直接失败,给你说下正确的依赖声明方式:
核心匹配规则
Python的tf-nightly版本号格式是X.Y.Z.devYYYYMMDD,对应的Android TensorFlow Lite Nightly依赖需要做两个关键调整:
- 把版本号里的
.替换成-,也就是改成X.Y.Z-devYYYYMMDD - 使用专用的Artifact ID:
tensorflow-lite-nightly(而非稳定版的tensorflow-lite加nightly后缀)
正确的依赖写法
对应你训练用的tf-nightly==2.4.0.dev20200817,build.gradle里的依赖应该写成:
implementation 'org.tensorflow:tensorflow-lite-nightly:2.4.0-dev20200817'
为什么之前的写法不对?
你尝试的2.4.0.dev20200817-nightly不符合TensorFlow Maven仓库的版本命名规范:
- Nightly版本的区分是通过Artifact ID(
tensorflow-lite-nightly)实现的,不需要在版本号末尾追加-nightly - 版本号里的
dev前缀需要和前面的版本号用-连接,而非.
这样修改后,Android端的TFLite版本就会和你训练用的tf-nightly版本完全对齐,应该就能解决推理兼容性问题了。
内容的提问来源于stack exchange,提问作者Joram
相关产品推荐
相关产品推荐

