解决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中的__getitem__专有方法
Jun 27 Python
Python如何发布程序的详细教程
Oct 09 Python
pandas.dataframe按行索引表达式选取方法
Oct 30 Python
Django框架model模型对象验证实现方法分析
Oct 02 Python
Pycharm 2020年最新激活码(亲测有效)
Sep 18 Python
python处理RSTP视频流过程解析
Jan 11 Python
PyTorch的SoftMax交叉熵损失和梯度用法
Jan 15 Python
Tensorflow安装问题: Could not find a version that satisfies the requirement tensorflow
Apr 20 Python
Expected conditions模块使用方法汇总代码解析
Aug 13 Python
python统计mysql数据量变化并调用接口告警的示例代码
Sep 21 Python
小白教你PyCharm从下载到安装再到科学使用PyCharm2020最新激活码
Sep 25 Python
关于Python OS模块常用文件/目录函数详解
Jul 01 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遍历数组的方法分享
2012/03/22 PHP
php实现用于验证所有类型的信用卡类
2015/03/24 PHP
心扬JS分页函数代码
2010/09/10 Javascript
通过JS来动态的修改url,实现对url的增删查改
2014/09/01 Javascript
JQuery判断radio是否选中并获取选中值的示例代码
2014/10/17 Javascript
表单元素值获取方式js及java方式的简单实例
2016/10/15 Javascript
解决微信内置浏览器返回上一页强制刷新问题方法
2017/02/05 Javascript
nodejs个人博客开发第二步 入口文件
2017/04/12 NodeJs
微信小程序实战之轮播图(3)
2017/04/17 Javascript
JavaScript限定范围拖拽及自定义滚动条应用(3)
2017/05/17 Javascript
angularjs+bootstrap实现自定义分页的实例代码
2017/06/19 Javascript
Vue.js 单页面多路由区域操作的实例详解
2017/07/17 Javascript
关于JavaScript的单双引号嵌套问题
2017/08/20 Javascript
基于jQuery的$.getScript方法去加载javaScript文档解析
2017/11/08 jQuery
深入理解Vue 组件之间传值
2018/08/16 Javascript
浅谈js闭包理解
2019/04/01 Javascript
解决layui动态添加的元素click等事件触发不了的问题
2019/09/20 Javascript
基于ts的动态接口数据配置的详解
2019/12/18 Javascript
python列表操作使用示例分享
2014/02/21 Python
给Python的Django框架下搭建的BLOG添加RSS功能的教程
2015/04/08 Python
在Django的URLconf中使用命名组的方法
2015/07/18 Python
Python如何快速实现分布式任务
2017/07/06 Python
在python里使用await关键字来等另外一个协程的实例
2020/05/04 Python
Python中pass的作用与使用教程
2020/11/13 Python
用python对excel查重
2020/12/07 Python
html5 Canvas画图教程(11)—使用lineTo/arc/bezierCurveTo画椭圆形
2013/01/09 HTML / CSS
html5+css3气泡组件的实现
2014/11/21 HTML / CSS
美国流行背包品牌:JanSport(杰斯伯)
2018/03/02 全球购物
行政管理人员精品工作推荐信
2013/11/04 职场文书
运动会致辞稿50字
2014/02/04 职场文书
求职毕业生自荐书
2014/02/08 职场文书
医院党员公开承诺书
2014/08/30 职场文书
2015关于重阳节的演讲稿
2015/03/20 职场文书
毕业论文指导老师意见
2015/06/04 职场文书
网络研修随笔感言
2015/11/18 职场文书
win10系统计算机图标怎么调出来?win10调出计算机图标的方法
2022/08/14 数码科技