时间:2021-05-22
我在做tensorflow开发的时候遇到如下的错误,我请教了深度学习社区q群中的大神,大神说这是运算图还在内存中,没有及时释放,需要restart一下kernel,但是由于我的代码不能停止执行,所以没办法用重新运行代码来解决释放内存运行图的问题。
问题:
with tf.Session() as sess:saver = tf.train.Saver() #保存运算图当我退出with并且立马执行
self.sess = tf.Session()self.saver = tf.train.import_meta_graph('./Model/model.ckpt.meta')self.saver.restore(self.sess, tf.train.latest_checkpoint('./Model/'))的时候由于session和graph没有释放内存的运算图就产生了报错,
解决办法:
在with tf.Session() as sess: 之后同时也要在with的范围以外(注意),添加
tf.reset_default_graph()代码来重置默认的图,这样就能解决我下一步执行代码
self.sess = tf.Session()self.saver = tf.train.import_meta_graph('./Model/model.ckpt.meta')self.saver.restore(self.sess, tf.train.latest_checkpoint('./Model/'))所产生的问题了。
以上这篇Tensorflow 实现释放内存就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。
声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。
C#中如何合理的释放非托管内存?在本文中我们将讲解使用IDisposable释放托管内存和非托管内存。 A.首先需要让类实现IDisposable接口,然
1.基本介绍tensorflow设备内存管理模块实现了一个best-fitwithcoalescing算法(后文简称bfc算法)。bfc算法是DoungLea'
前言本文告诉大家如何使用Marshal做出可以快速释放内存的大数组。最近在做3D,需要不断申请一段大内存数组,然后就释放他,但是C#对于大内存不是立刻释放,所以
在用tensorflow实现一些模型的时候,有时候我们在运行程序的时候,会发现程序占用的内存在不断增长。最后内存溢出,程序被kill掉了。这个问题,其实有两个可
1.查看内存使用情况free-g当观察到free栏已为0的时候,表示内存基本被吃完了,那就释放内存吧。释放内存:syncecho3>/proc/sys/vm/d