我已经看到了限制TensorFlows使用GPU内存的方法(通过比率)。我们在移动设备上运行TF,RAM非常稀缺。有没有办法影响TF的内存分配?我必须以某种方式确保TF在任何时候都不会分配超过400MB的空间,否则应用程序将崩溃。
TF分配特别是在运行期间达到峰值,并在(594MB峰值,355MB空闲)之后回落到合理水平。
此外,是否有任何方法可以优化低内存分配图?
PS:在TF 1.0中,内存分配似乎已大幅增加,我不记得它在0.9 beta的情况下是如此高,特别是当图表在磁盘上只有45MB时。