解决Tensorflow占用GPU显存问题


Posted in Python onFebruary 03, 2020

我使用Pytorch进行模型训练时发现真正模型本身对于显存的占用并不明显,但是对应的转换为tensorflow后(权重也进行了转换),发现Python-tensorflow在使用时默认吃掉所有显存,并且不手动终结程序的话显存并不释放(我有两个序贯的模型,前面一个跑完后并不释放占用显存)(https://github.com/tensorflow/tensorflow/issues/1727),这一点对于后续的工作有很大的影响。

后面发现python-tensorflow限制显存有两种方法:

1. 设置显卡的使用率

这种方法在学习和工作中比较好用,学习时可提高显卡使用效率,工作时可方便的获得GPU显存消耗极限,用以提供显卡购买时的参数,现将代码展示如下:

解决Tensorflow占用GPU显存问题

这里的0.1 表示使用显存总量的的10%

2. 设置显卡按需使用(这个本人并没有专门测试,只是从tensorflow论坛上获得)

gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

以上这篇解决Tensorflow占用GPU显存问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
在Python中使用M2Crypto模块实现AES加密的教程
Apr 08 Python
python变量命名的7条建议
Jul 04 Python
解决yum对python依赖版本问题
Jul 05 Python
django 做 migrate 时 表已存在的处理方法
Aug 31 Python
Python字符串中添加、插入特定字符的方法
Sep 10 Python
python 哈希表实现简单python字典代码实例
Sep 27 Python
python脚本实现音频m4a格式转成MP3格式的实例代码
Oct 09 Python
matplotlib.pyplot画图并导出保存的实例
Dec 07 Python
python 回溯法模板详解
Feb 26 Python
Python常用编译器原理及特点解析
Mar 23 Python
pycharm2020.1.2永久破解激活教程,实测有效
Oct 29 Python
Python爬虫后获取重定向url的两种方法
Jan 19 Python
通过Turtle库在Python中绘制一个鼠年福鼠
Feb 03 #Python
python爬虫模块URL管理器模块用法解析
Feb 03 #Python
Tensorflow实现多GPU并行方式
Feb 03 #Python
python如何通过twisted搭建socket服务
Feb 03 #Python
关于Tensorflow分布式并行策略
Feb 03 #Python
基于python修改srt字幕的时间轴
Feb 03 #Python
Python实现不规则图形填充的思路
Feb 02 #Python
You might like
PHP中对数据库操作的封装
2006/10/09 PHP
php利用新浪接口查询ip获取地理位置示例
2014/01/20 PHP
php解决和避免form表单重复提交的几种方法
2016/08/31 PHP
PHP-X系列教程之内置函数的使用示例
2017/10/16 PHP
php 实现简单的登录功能示例【基于thinkPHP框架】
2019/12/02 PHP
swoole锁的机制代码实例讲解
2021/03/04 PHP
Dojo之路:如何利用Dojo实现Drag and Drop效果
2007/04/10 Javascript
该如何加载google-analytics(或其他第三方)的JS
2010/05/13 Javascript
浅析Node.js查找字符串功能
2014/09/03 Javascript
JavaScript常用的弹出广告及背投广告实现方法
2015/02/06 Javascript
javascript动态获取登录时间和在线时长
2016/02/25 Javascript
js实现音频控制进度条功能
2017/04/01 Javascript
利用babel将es6语法转es5的简单示例
2017/12/01 Javascript
JavaScript实现计算多边形质心的方法示例
2018/01/31 Javascript
Node.js使用Angular简单示例
2018/05/11 Javascript
python pygame实现2048游戏
2018/11/20 Python
python 导入数据及作图的实现
2019/12/03 Python
pandas分组聚合详解
2020/04/10 Python
Python新手如何进行闭包时绑定变量操作
2020/05/29 Python
Python爬虫破解登陆哔哩哔哩的方法
2020/11/17 Python
SEPHORA丝芙兰捷克官网:购买香水、化妆品和护肤品
2018/11/26 全球购物
Eagle Eyes Optics鹰眼光学:高性能太阳镜
2018/12/07 全球购物
专注澳大利亚特产和新西兰特产的澳洲中文网:0061澳洲制造
2019/03/24 全球购物
华为c/c++笔试题
2016/01/25 面试题
高中毕业生自我鉴定范文
2013/09/26 职场文书
公司董事长职责
2013/12/12 职场文书
校园广播稿500字
2014/02/04 职场文书
《鲁班和橹板》教学反思
2014/04/27 职场文书
竞选学习委员演讲稿
2014/04/28 职场文书
党的群众路线教育实践活动方案
2014/10/31 职场文书
维稳承诺书
2015/01/20 职场文书
担保贷款承诺书
2015/04/30 职场文书
学生会部长竞选稿
2015/11/19 职场文书
《葡萄沟》教学反思
2016/02/23 职场文书
python - timeit 时间模块
2021/04/06 Python
vue backtop组件的实现完整代码
2021/04/07 Vue.js