请问一下有什么框架可以实现对深度学习多模型的自动显存管理

2021-05-21 16:14:56 +08:00
 AlwaysPersist

目前有一个项目,包含超过 40 个深度学习模型,但是显存无法容纳所有的模型,因此想问下,有没有什么框架,可以实现对预模型的动态加载。主要是 pytorch 类型的深度学习模型。

880 次点击
所在节点    Python
0 条回复

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

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

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

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

© 2021 V2EX