微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

tensorflow 2 推理期间的高 CPU 内存使用兼容模式

如何解决tensorflow 2 推理期间的高 CPU 内存使用兼容模式

我使用 .pb 格式的预训练类 ENET 模型对流量进行图像分割,它占用了我 1.8 GB 的 RAM。

我知道深度学习模型会占用大量 VRAM 空间,但这是我第一次看到模型在我的 cpu 上使用这么多内存。我试图限制 GPU 上的内存,但它对 RAM 使用没有任何影响。有人遇到过这个问题吗?

规格:16GB RAM,Geforce GTX 1050 2GB VRAM,tensorflow 2.3.1 我还对兼容模式进行了推理。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。