【发布时间】：2019-02-21 07:36:29
【问题描述】：
有什么方法可以防止为 MXNet 分配完整的 GPU 内存？所以它只分配它需要的东西，而不是整个 GPU 内存。
我想在 MXNet 旁边的同一个 GPU 上使用 Tensorflow/Keras 中的另一个模型，似乎整个内存都被 MXNet 保留了。
【问题讨论】：
-
你找到解决办法了吗？
标签： mxnet
【发布时间】：2019-02-21 07:36:29
【问题描述】：
MXNet 分配内存as needed。也许您的程序中存在内存泄漏，或者 Tensorflow 正在尝试在整个 GPU 上预分配内存，即default behavior。该行为可使用tf.GPUOptions 进行配置。请参阅有关如何使用这些选项的链接。
希望对您有所帮助， 维沙尔
【讨论】：