Tensorflow 实现释放内存


Posted in Python onFebruary 03, 2020

我在做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没有释放内存的运算图就产生了报错,

Tensorflow 实现释放内存

解决办法:

在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 相关文章推荐
举例讲解Python中的算数运算符的用法
May 13 Python
Python常用算法学习基础教程
Apr 13 Python
PyChar学习教程之自定义文件与代码模板详解
Jul 17 Python
Python爬虫实战:分析《战狼2》豆瓣影评
Mar 26 Python
Pycharm代码无法复制,无法选中删除,无法编辑的解决方法
Oct 22 Python
Python 最强编辑器详细使用指南(PyCharm )
Sep 16 Python
Pandas操作CSV文件的读写实现方法
Nov 13 Python
使用pandas实现连续数据的离散化处理方式(分箱操作)
Nov 22 Python
python socket通信编程实现文件上传代码实例
Dec 14 Python
django xadmin action兼容自定义model权限教程
Mar 30 Python
python+selenium实现12306模拟登录的步骤
Jan 21 Python
matplotlib之pyplot模块之标题(title()和suptitle())
Feb 22 Python
手把手教你进行Python虚拟环境配置教程
Feb 03 #Python
解决TensorFlow GPU版出现OOM错误的问题
Feb 03 #Python
python global和nonlocal用法解析
Feb 03 #Python
Tensorflow设置显存自适应,显存比例的操作
Feb 03 #Python
解决Tensorflow占用GPU显存问题
Feb 03 #Python
通过Turtle库在Python中绘制一个鼠年福鼠
Feb 03 #Python
python爬虫模块URL管理器模块用法解析
Feb 03 #Python
You might like
PHP 读取Postgresql中的数组
2013/04/14 PHP
PHP删除二维数组中相同元素及数组重复值的方法示例
2017/05/05 PHP
PHP实现登陆并抓取微信列表中最新一组微信消息的方法
2017/07/10 PHP
详谈PHP中public,private,protected,abstract等关键字的用法
2017/12/31 PHP
JavaScript confirm选择判断
2008/10/18 Javascript
JavaScript Accessor实现说明
2010/12/06 Javascript
jquery利用ajax调用后台方法实例
2013/08/23 Javascript
jquery 隐藏与显示tr标签示例代码
2014/06/06 Javascript
微信小程序 Toast自定义实例详解
2017/01/20 Javascript
React.js中常用的ES6写法总结(推荐)
2017/05/09 Javascript
原生js中ajax访问的实例详解
2017/09/19 Javascript
JS+canvas画一个圆锥实例代码
2017/12/13 Javascript
setTimeout时间设置为0详细解析
2018/03/13 Javascript
jQuery实现图片上传预览效果功能完整实例【测试可用】
2018/05/28 jQuery
实例详解vue中的$root和$parent
2019/04/29 Javascript
vue + typescript + video.js实现 流媒体播放 视频监控功能
2019/07/07 Javascript
详解element-ui级联菜单(城市三级联动菜单)和回显问题
2019/10/02 Javascript
vue中在vuex的actions中请求数据实例
2019/11/08 Javascript
在js文件中引入(调用)另一个js文件的三种方法
2020/09/11 Javascript
python中定义结构体的方法
2013/03/04 Python
Python编程flask使用页面模版的方法
2018/12/28 Python
Python实现直方图均衡基本原理解析
2019/08/08 Python
安装python及pycharm的教程图解
2019/10/10 Python
python基于plotly实现画饼状图代码实例
2019/12/16 Python
python手写均值滤波
2020/02/19 Python
Jupyter notebook命令和编辑模式常用快捷键汇总
2020/11/17 Python
猫途鹰英国网站:TripAdvisor英国(旅游社区和旅游评论)
2016/08/30 全球购物
JSF面试题:如何管量web层中的Bean,用什么标签。如何通过jsp页面与Bean绑定在一起进行处理?
2012/10/05 面试题
七一党建活动方案
2014/01/28 职场文书
顶碗少年教学反思
2014/02/21 职场文书
开学典礼策划方案
2014/05/28 职场文书
2014年领导班子专项整治整改方案
2014/09/28 职场文书
法院干警四风问题自我剖析材料
2014/09/29 职场文书
学校纪律作风整改措施思想汇报
2014/10/11 职场文书
考试作弊检讨书
2015/01/27 职场文书
销售口号霸气押韵
2015/12/24 职场文书