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

相关文章

  • Python3爬虫教程之利用Python实现发送天气预报邮件

    Python3爬虫教程之利用Python实现发送天气预报邮件

    这篇文章主要给大家介绍了关于Python3爬虫教程之利用Python实现发送天气预报邮件的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面来一起看看吧
    2018-12-12
  • 33个Python爬虫项目实战(推荐)

    33个Python爬虫项目实战(推荐)

    这篇文章主要介绍了33个Python爬虫项目实战,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2019-07-07
  • Python线程之定位与销毁的实现

    Python线程之定位与销毁的实现

    这篇文章主要介绍了Python线程之定位与销毁的实现,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2019-02-02
  • Matplotlib子图的创建的实现

    Matplotlib子图的创建的实现

    本文主要介绍了Matplotlib子图的创建的实现,包括fig.add_axes()创建子图和plt.axes创建子图这两种方法,具有一定的参考价值,感兴趣的可以了解一下
    2023-11-11
  • pyqt5 QScrollArea设置在自定义侧(任何位置)

    pyqt5 QScrollArea设置在自定义侧(任何位置)

    这篇文章主要介绍了pyqt5 QScrollArea设置在自定义侧(任何位置),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-09-09
  • 利用python编写一个图片主色转换的脚本

    利用python编写一个图片主色转换的脚本

    这篇文章主要给大家介绍了关于利用python编写一个图片主色转换脚本的相关资料,主要使用的是Python中的Pillow图像处理库,文中通过示例代码介绍的非常详细,需要的朋友可以参考借鉴,下面随着小编来一起看看吧。
    2017-12-12
  • 用Python代码来绘制彭罗斯点阵的教程

    用Python代码来绘制彭罗斯点阵的教程

    这篇文章主要介绍了用Python代码来绘制彭罗斯点阵的教程,核心代码主要就是一行简单的lambda匿名函数,需要的朋友可以参考下
    2015-04-04
  • 用Python获取摄像头并实时控制人脸的实现示例

    用Python获取摄像头并实时控制人脸的实现示例

    这篇文章主要介绍了用Python获取摄像头并实时控制人脸的实现示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-07-07
  • Redis持久化机制实现原理和流程

    Redis持久化机制实现原理和流程

    这篇文章主要介绍了Redis持久化机制实现原理和流程,持久化就是把内存中的数据存放到磁盘中,防止宕机后内存数据丢失,需要的朋友可以参考下
    2023-04-04
  • python rpyc客户端调用服务端方法的注意说明

    python rpyc客户端调用服务端方法的注意说明

    这篇文章主要介绍了python rpyc客户端调用服务端方法的注意说明,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2022-06-06

最新评论