解决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环境下安装使用异步任务队列包Celery的基础教程
May 07 Python
解决PyCharm中光标变粗的问题
Aug 05 Python
python字典DICT类型合并详解
Aug 17 Python
教你用 Python 实现微信跳一跳(Mac+iOS版)
Jan 04 Python
Python实现将MySQL数据库表中的数据导出生成csv格式文件的方法
Jan 11 Python
Windows上使用Python增加或删除权限的方法
Apr 24 Python
Python读写zip压缩文件的方法
Aug 29 Python
python查看模块,对象的函数方法
Oct 16 Python
python爬取盘搜的有效链接实现代码
Jul 20 Python
使用Python和OpenCV检测图像中的物体并将物体裁剪下来
Oct 30 Python
Python使用matplotlib绘制Logistic曲线操作示例
Nov 28 Python
什么是python的列表推导式
May 26 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一些常用的正则表达式字符的一些转换
2008/07/29 PHP
php获取网卡的MAC地址支持WIN/LINUX系统
2014/04/30 PHP
Yii2使用表单上传文件的实例代码
2017/08/03 PHP
js tab效果的实现代码
2009/12/26 Javascript
利用location.hash实现跨域iframe自适应
2010/05/04 Javascript
JavaScript 异步方法队列链实现代码分析
2010/06/05 Javascript
Js event事件在IE、FF兼容性问题
2011/01/01 Javascript
JavaScript中变量提升 Hoisting
2012/07/03 Javascript
node.js中的querystring.unescape方法使用说明
2014/12/10 Javascript
js中函数声明与函数表达式
2015/06/03 Javascript
jQuery实现的Tab滑动选项卡及图片切换(多种效果)小结
2015/09/14 Javascript
纯JavaScript代码实现移动设备绘图解锁
2015/10/16 Javascript
简单实现JS上传图片预览功能
2017/04/14 Javascript
JS简单实现点击跳转登陆邮箱功能的方法
2017/10/31 Javascript
10行原生JS实现文字无缝滚动(超简单)
2018/01/02 Javascript
axios使用拦截器统一处理所有的http请求的方法
2018/11/02 Javascript
vue插件draggable实现拖拽移动图片顺序
2018/12/01 Javascript
深入理解Vue.js轻量高效的前端组件化方案
2018/12/10 Javascript
VUE中setTimeout和setInterval自动销毁案例
2020/09/07 Javascript
python实现从字典中删除元素的方法
2015/05/04 Python
通过Python来使用七牛云存储的方法详解
2015/08/07 Python
Python中反射和描述器总结
2018/09/23 Python
通过python实现windows桌面截图代码实例
2020/01/17 Python
使用python图形模块turtle库绘制樱花、玫瑰、圣诞树代码实例
2020/03/16 Python
Python爬虫+tkinter界面实现历史天气查询的思路详解
2021/02/22 Python
Perfume’s Club法国站:购买香水和化妆品
2019/05/02 全球购物
Ramy Brook官网:美国现代女装品牌
2019/06/18 全球购物
外贸公司实习自我鉴定
2013/09/24 职场文书
电子信息毕业生自荐信
2013/11/16 职场文书
《雾凇》教学反思
2014/02/17 职场文书
师德自我剖析材料范文
2014/10/06 职场文书
2014年幼儿园教师工作总结
2014/11/08 职场文书
写给老师的感谢信
2015/01/20 职场文书
居安思危观后感
2015/06/11 职场文书
教你用Java在个人电脑上实现微信扫码支付
2021/06/13 Java/Android
python+pyhyper实现识别图片中的车牌号思路详解
2022/12/24 Python