限制TensorFlow RAM的使用

时间:2017-03-15 14:22:53

标签: c++ tensorflow

我已经看到了限制TensorFlows使用GPU内存的方法(通过比率)。我们在移动设备上运行TF,RAM非常稀缺。有没有办法影响TF的内存分配?我必须以某种方式确保TF在任何时候都不会分配超过400MB的空间,否则应用程序将崩溃。

TF分配特别是在运行期间达到峰值,并在(594MB峰值,355MB空闲)之后回落到合理水平。

此外,是否有任何方法可以优化低内存分配图?

PS:在TF 1.0中,内存分配似乎已大幅增加,我不记得它在0.9 beta的情况下是如此高,特别是当图表在磁盘上只有45MB时。

0 个答案:

没有答案