解决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使用scrapy采集数据时为每个请求随机分配user-agent的方法
Apr 08 Python
Python导出数据到Excel可读取的CSV文件的方法
May 12 Python
浅谈python新手中常见的疑惑及解答
Jun 14 Python
Python模拟登陆淘宝并统计淘宝消费情况的代码实例分享
Jul 04 Python
python利用有道翻译实现"语言翻译器"的功能实例
Nov 14 Python
python3+PyQt5实现文档打印功能
Apr 24 Python
python Qt5实现窗体跟踪鼠标移动
Dec 13 Python
python三引号如何输入
Jul 06 Python
Pycharm如何自动生成头文件注释
Nov 14 Python
Opencv 图片的OCR识别的实战示例
Mar 02 Python
Pytorch实现图像识别之数字识别(附详细注释)
May 11 Python
python中pd.cut()与pd.qcut()的对比及示例
Jun 16 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 chr() ord()中文截取乱码问题解决方法
2008/09/08 PHP
浅析PHP中的UNICODE 编码与解码
2013/06/29 PHP
PHP不用递归实现无限分级的例子分享
2014/04/18 PHP
php堆排序实现原理与应用方法
2015/01/03 PHP
JavaScript的漂亮的代码片段
2013/06/05 Javascript
让新消息在网页标题闪烁提示的jQuery代码
2013/11/04 Javascript
jquery 操作iframe的几种方法总结
2013/12/13 Javascript
基于JavaScript实现全屏透明遮罩div层锁屏效果
2016/01/26 Javascript
js实现的简单图片浮动效果完整实例
2016/05/10 Javascript
解析jquery easyui tree异步加载子节点问题
2017/03/08 Javascript
laravel5.3 vue 实现收藏夹功能实例详解
2018/01/21 Javascript
Vue中父子组件通讯之todolist组件功能开发
2018/05/21 Javascript
详解vue.js下引入百度地图jsApi的两种方法
2018/07/27 Javascript
vue实现word,pdf文件的导出功能
2018/07/31 Javascript
vue 本地环境跨域请求proxyTable的方法
2018/09/19 Javascript
Vue form表单动态添加组件实战案例
2019/09/02 Javascript
将Python中的数据存储到系统本地的简单方法
2015/04/11 Python
Python使用Mechanize模块编写爬虫的要点解析
2016/03/31 Python
使用python绘制常用的图表
2016/08/27 Python
python的random模块及加权随机算法的python实现方法
2017/01/04 Python
python 编码规范整理
2018/05/05 Python
python读取目录下所有的jpg文件,并显示第一张图片的示例
2019/06/13 Python
使用Pyhton集合set()实现成果查漏的例子
2019/11/24 Python
keras获得model中某一层的某一个Tensor的输出维度教程
2020/01/24 Python
python反爬虫方法的优缺点分析
2020/11/25 Python
使用numpngw和matplotlib生成png动画的示例代码
2021/01/24 Python
建筑专业毕业生推荐信
2013/11/21 职场文书
《广玉兰》教学反思
2014/04/14 职场文书
幼儿教师师德承诺书
2014/05/23 职场文书
教师求职自荐书
2014/06/14 职场文书
大学生毕业评语
2014/12/31 职场文书
优秀团员个人总结
2015/02/26 职场文书
2015财务年度工作总结范文
2015/05/04 职场文书
python文件目录操作之os模块
2021/05/08 Python
pandas 实现将NaN转换为None
2021/05/14 Python
golang 语言中错误处理机制
2021/08/30 Golang