首页 > 解决方案 > 如何在 500MiB 的免费 slug 大小内在 Heroku 中部署 tensorflow 模型?

问题描述

我有一个用 tensorflow 2.6 制作的卷积神经网络模型。我希望以免费的 Heroku 计划部署它。但是 Heroku 免费计划仅允许 500MiB 的最大 slug 大小。有什么方法可以在 Heroku 免费计划中托管模型?

我的最终目标是构建一个 API,当与 base64 格式的图像一起传递时,它会返回一个标签。

标签: pythontensorflowheroku

解决方案


我找到了解决方案。

使用tensorflow-cpu.

tensorflow-cpu比 Heroku 免费的最大允许大小 (500MiB ) 小得多,tensorflow并且可以放入其中。


推荐阅读