tensorflow引起的内存泄露

计算机配置内存12G,显存4G,运行有10分钟左右就提醒说内存不够用然后退出运行,直觉是程序bug问题
终于找到靠谱解决方案:
http://cherishlc.iteye.com/blog/2324796
这个博客的第二个给了大致的讲解,在给出的附录中:
https://stackoverflow.com/documentation/tensorflow/3883/how-to-debug-a-memory-leak-in-tensorflow/13426/use-graph-finalize-to-catch-nodes-being-added-to-the-graph#t=201707221208374841351
讲解更详细,并给出了解决的方法
即:sess.graph.finalize() 使得整个graph变为只读的,不能再向图中添加任何节点
我也明白了我的错误原因,在for循环中用了tf.convert_to_tensor方法,相当于不停地向图中添加节点,谢谢给出了解决方案的大牛们
另一个附录:https://github.com/tensorflow/tensorflow/issues/4151
通过一个示例给出了可行的替换方法,非常具有可行性

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值