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中优化NumPy包使用性能的教程
Apr 23 Python
Python实现将文本生成二维码的方法示例
Jul 18 Python
python调用百度语音REST API
Aug 30 Python
Python脚本完成post接口测试的实例
Dec 17 Python
python配置grpc环境
Jan 01 Python
python环境路径配置以及命令行运行脚本
Apr 02 Python
关于Python形参打包与解包小技巧分享
Aug 24 Python
Python3中configparser模块读写ini文件并解析配置的用法详解
Feb 18 Python
Python实现RabbitMQ6种消息模型的示例代码
Mar 30 Python
python实现密码验证合格程序的思路详解
Jun 01 Python
pytorch快速搭建神经网络_Sequential操作
Jun 17 Python
python基础之函数的定义和调用
Oct 24 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的库,结果发现很多东西
2006/12/31 PHP
Pain 全世界最小最简单的PHP模板引擎 (普通版)
2011/10/23 PHP
PHP递归删除目录几个代码实例
2014/04/21 PHP
PHP.ini安全配置检测工具pcc简单介绍
2015/07/02 PHP
PHP 在数组中搜索给定的简单实例 array_search 函数
2016/06/13 PHP
php中10个不同等级压缩优化图片操作示例
2016/11/14 PHP
thinkphp ajaxfileupload实现异步上传图片的示例
2017/08/28 PHP
PHP实现QQ、微信和支付宝三合一收款码实例代码
2018/02/19 PHP
PHP中引用类型和值类型功能与用法示例
2019/02/26 PHP
JQuery读取XML文件数据并显示的实现代码
2009/12/16 Javascript
再谈javascript 动态添加样式规则 W3C校检
2009/12/25 Javascript
jquery load事件(callback/data)使用方法及注意事项
2013/02/06 Javascript
详解jQuery插件开发中的extend方法
2013/11/19 Javascript
将数字转换成大写的人民币表达式的js函数
2014/09/21 Javascript
基于NodeJS的前后端分离的思考与实践(三)轻量级的接口配置建模框架
2014/09/26 NodeJs
node.js中的events.emitter.removeAllListeners方法使用说明
2014/12/10 Javascript
JavaScript生成验证码并实现验证功能
2016/09/24 Javascript
jQuery grep()方法详解及实例代码
2016/10/30 Javascript
Javascript实现信息滚动效果
2017/05/18 Javascript
jQuery实现的淡入淡出与滑入滑出效果示例
2018/04/18 jQuery
对Vue beforeRouteEnter 的next执行时机详解
2018/08/25 Javascript
swiperjs实现导航与tab页的联动
2020/12/13 Javascript
[54:27]TNC vs Serenity 2018国际邀请赛小组赛BO2 第一场 8.18
2018/08/19 DOTA
python判断字符串是否是json格式方法分享
2017/11/07 Python
让Django支持Sql Server作后端数据库的方法
2018/05/29 Python
python简单实现矩阵的乘,加,转置和逆运算示例
2019/07/10 Python
python web框架中实现原生分页
2019/09/08 Python
用什么库写 Python 命令行程序(示例代码详解)
2020/02/20 Python
CSS3 input框的实现代码类似Google登录的动画效果
2020/08/04 HTML / CSS
欧洲品牌瓷器餐具网上商店:Porzellantreff.de
2018/04/04 全球购物
财务部总监岗位职责
2014/03/12 职场文书
领导参观欢迎词
2015/01/26 职场文书
幼儿教师师德师风自我评价
2015/03/05 职场文书
环保宣传语大全
2015/07/13 职场文书
医院病假条怎么写
2015/08/17 职场文书
Python面试不修改数组找出重复的数字
2022/05/20 Python