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模拟登陆阿里妈妈生成商品推广链接
Apr 03 Python
Python开发WebService系列教程之REST,web.py,eurasia,Django
Jun 30 Python
Python argv用法详解
Jan 08 Python
用TensorFlow实现多类支持向量机的示例代码
Apr 28 Python
Python连接Redis的基本配置方法
Sep 13 Python
Python3编码问题 Unicode utf-8 bytes互转方法
Oct 26 Python
对pandas数据判断是否为NaN值的方法详解
Nov 06 Python
Django数据库连接丢失问题的解决方法
Dec 29 Python
Selenium+Python 自动化操控登录界面实例(有简单验证码图片校验)
Jun 28 Python
python读取图片的方式,以及将图片以三维数组的形式输出方法
Jul 03 Python
Django 项目重命名的实现步骤解析
Aug 14 Python
详解python polyscope库的安装和例程
Nov 13 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
javascript some()函数用法详解
2014/11/13 PHP
谈谈PHP中substr和substring的正确用法及相关参数的介绍
2015/12/16 PHP
Thinkphp自定义代码生成工具及用法说明(附下载地址)
2016/05/27 PHP
PDO的安全处理与事物处理方法
2016/10/31 PHP
PHP 实现文件压缩解压操作的方法
2019/06/14 PHP
使用JavaScript switch case 另类写法
2010/03/14 Javascript
JS中的log对象获取以及debug的写法介绍
2014/03/03 Javascript
jQuery焦点图切换简易插件制作过程全纪录
2014/08/27 Javascript
js实现最短的XML格式化工具实例
2015/03/12 Javascript
用JavaScript判断CSS浏览器类型前缀的两种方法
2015/10/08 Javascript
深入探讨javascript函数式编程
2015/10/11 Javascript
js判断当前页面用什么浏览器打开的方法
2016/01/06 Javascript
Highcharts学习之坐标轴
2016/08/02 Javascript
jQuery实现微信长按识别二维码功能
2016/08/26 Javascript
JavaScript用构造函数如何获取变量的类型名
2016/12/23 Javascript
js验证手机号、密码、短信验证码代码工具类
2020/06/24 Javascript
jQuery Chosen通用初始化
2017/03/07 Javascript
jQuery Ajax自定义分页组件(jquery.loehpagerv1.0)实例详解
2017/05/01 jQuery
javaScript中"=="和"==="的区别详解
2018/03/16 Javascript
用npm安装vue和vue-cli,并使用webpack创建项目的方法
2018/09/28 Javascript
AngularJS 多指令Scope问题的解决
2018/10/25 Javascript
jQuery实现的导航条点击后高亮显示功能示例
2019/03/04 jQuery
nodejs检测因特网是否断开的解决方案
2019/04/17 NodeJs
Python实现3行代码解简单的一元一次方程
2014/08/18 Python
python安装教程 Pycharm安装详细教程
2017/05/02 Python
pycharm 将django中多个app放到同个文件夹apps的处理方法
2018/05/30 Python
Python Django给admin添加Action的方法实例详解
2019/04/29 Python
对Django 中request.get和request.post的区别详解
2019/08/12 Python
Speedo速比涛法国官方网站:泳衣、泳镜、泳帽、泳裤
2019/07/30 全球购物
大学生党课思想汇报
2013/12/29 职场文书
服务承诺书范文
2014/05/19 职场文书
酒店周年庆活动方案
2014/08/21 职场文书
学习党的群众路线教育实践活动剖析材料
2014/10/13 职场文书
Python中的xlrd模块使用整理
2021/06/15 Python
剑指Offer之Java算法习题精讲二叉树专项训练
2022/03/21 Java/Android
python使用torch随机初始化参数
2022/03/22 Python