问MXNet -如何防止满内存分配
EN

Stack Overflow用户

提问于 2019-02-21 15:36:30

回答 1查看 115关注 0票数 2

有没有办法防止为MXNet分配全部的GPU内存？因此它只分配它需要的内存，而不是整个GPU内存。

我想在与MXNet相同的图形处理器上使用Tensorflow/ MXNet中的另一个模型，似乎整个内存都被GPU保留了。

回答已采纳

发布于 2019-03-06 05:23:21

MXNet分配内存as needed。也许你的程序中存在内存泄漏，或者Tensorflow试图在整个图形处理器(即default behavior )上预先分配内存。该行为可以使用tf.GPUOptions进行配置。请参阅有关如何使用这些选项的链接。

希望这能帮上忙，Vishaal

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/54801596

复制

相似问题

问MXNet -如何防止满内存分配EN