有什么方法可以防止为MXNet分配完整的GPU内存吗？这样它只分配所需的内容，而不分配整个GPU的内存。
我想与MXNet一起在同一GPU上在Tensorflow / Keras中使用另一种模型，并且似乎整个内存都由MXNet保留。
答案 0 :(得分：1)
MXNet分配内存as needed。程序中可能存在内存泄漏，或者Tensorflow尝试在整个GPU（default behavior）上预分配内存。该行为可以通过tf.GPUOptions进行配置。请参阅有关如何使用这些选项的链接。
希望有帮助， Vishaal