请教 docker TensorFlow/Saving，部署模型的时候怎样配置或者判断显存应该是多少。

2020-05-07 22:57:25 +08:00
 naldo0193

学习使用 docker TensorFlow/Saving 部署，显存满了，实际上应该用不了这么多显存（猜的）。使用 docker TensorFlow/Saving 可以配置显存大小吗？有没有方法去预估部署的模型应该是多少显存呢？

1346 次点击
所在节点    问与答
3 条回复
c0xt30a
2020-05-07 23:36:21 +08:00
TF 默认占用所有显存。

如果运行时不足，可以试着
1. 修改 batch size
2. 如果是图片之类的数据，divide and conquer
joyme
2020-05-07 23:44:11 +08:00
你想打的是 tensorflow serving 吗？有个这个参数可以试试：--per_process_gpu_memory_fraction=0.5
naldo0193
2020-05-08 09:00:43 +08:00
@c0xt30a
@joyme
感谢

这是一个专为移动设备优化的页面（即为了让你能够在 Google 搜索结果里秒开这个页面），如果你希望参与 V2EX 社区的讨论，你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/669452

V2EX 是创意工作者们的社区，是一个分享自己正在做的有趣事物、交流想法，可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX