我已经看到了限制TensorFlows使用GPU内存的方法(通过比率).我们在移动设备上运行TF,RAM非常稀缺.有没有办法影响TF的内存分配？我必须以某种方式确保TF在任何时候都不会分配超过400MB,否则应用程序将崩溃.
TF分配尤其在运行期间达到峰值,并且在(594MB峰值,355MB空闲)之后回落到合理水平.
此外,是否有任何方法可以优化图形以实现低内存分配？
PS:看起来内存分配已经随着TF 1.0大幅增加,我不记得它与0.9 beta这么高,特别是当图表在磁盘上只有45MB时.
| 归档时间： |
|
| 查看次数： |
707 次 |
| 最近记录： |