Tensorflow 实现释放内存

更新时间：2020年02月03日 10:17:03 作者：荣轩浩

今天小编就为大家分享一篇Tensorflow 实现释放内存，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

我在做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 实现释放内存就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python之csv文件从MySQL数据库导入导出的方法
今天小编就为大家分享一篇Python之csv文件从MySQL数据库导入导出的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-06-06
Python中的jquery PyQuery库使用小结
这篇文章主要介绍了Python中的jquery PyQuery库使用小结,需要的朋友可以参考下
2014-05-05
python 淘宝爬虫小实例
双十一即将到来，电商都在做活动打折，但打完折是不是真的优惠了，需要我们自己斟酌，毕竟我们不能一直关注着价格，也自然不能知道现在的价格比以前高了还是低了，今天让我们用Python来爬取一下淘宝吧
2021-11-11
在Python中使用AOP实现Redis缓存示例
本篇文章主要介绍了在Python中使用AOP实现Redis缓存示例，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2017-07-07
详细讲解用Python发送SMTP邮件的教程
这篇文章主要详细讲解了用Python发送SMTP邮件的教程,包括在邮件中添加图片等文件,强烈推荐!需要的朋友可以参考下
2015-04-04
Python学习小技巧之列表项的推导式与过滤操作
这篇文章主要给大家介绍了Python学习小技巧之列表项的推导式与过滤操作的相关资料，文中介绍的非常详细，对大家具有一定的参考学习价值，需要的朋友们下面来一起看看把。
2017-05-05
flask-script模块的具体使用
本文主要介绍了flask-script模块的具体使用，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-11-11
python字典多条件排序方法实例
这篇文章主要介绍了python字典多条件排序方法实例,分别使用lambda和itemgetter实现,需要的朋友可以参考下
2014-06-06
python数字图像处理之高级滤波代码详解
这篇文章主要介绍了python数字图像处理之高级滤波代码详解，介绍了许多对图像处理的滤波方法，具有一定参考价值，需要的朋友可以了解下。
2017-11-11
深入浅析python with语句简介
with 语句适用于对资源进行访问的场合，确保不管使用过程中是否发生异常都会执行必要的“清理”操作，释放资源，这篇文章给大家介绍了python with语句简介，感兴趣的朋友一起看看吧
2018-04-04