tensorflow 限制显存大小的实现


Posted in Python onFebruary 03, 2020

Python在用GPU跑模型的时候最好开多进程,因为很明显这种任务就是计算密集型的。

用进程池好管理,但是tensorflow默认情况会最大占用显存,尽管该任务并不需要这么多,因此我们可以设置显存的按需获取,这样程序就不会死掉了。

1. 按比例预留:

tf_config = tensorflow.ConfigProto() 
tf_config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 分配50% 
session = tensorflow.Session(config=tf_config)

2. 或者干脆自适应然后自动增长:

tf_config = tensorflow.ConfigProto() 
tf_config.gpu_options.allow_growth = True # 自适应 
session = tensorflow.Session(config=tf_config)

以上这篇tensorflow 限制显存大小的实现就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中用max()方法求最大值的介绍
May 15 Python
python简单获取数组元素个数的方法
Jul 13 Python
利用Python暴力破解zip文件口令的方法详解
Dec 21 Python
python修改txt文件中的某一项方法
Dec 29 Python
Python设计模式之享元模式原理与用法实例分析
Jan 11 Python
python  logging日志打印过程解析
Oct 22 Python
python多线程案例之多任务copy文件完整实例
Oct 29 Python
tensorflow对图像进行拼接的例子
Feb 05 Python
Python之Matplotlib文字与注释的使用方法
Jun 18 Python
python关于倒排列的知识点总结
Oct 13 Python
python list等分并从等分的子集中随机选取一个数
Nov 16 Python
Django与数据库交互的实现
Jun 03 Python
基于tensorflow指定GPU运行及GPU资源分配的几种方式小结
Feb 03 #Python
tensorflow指定GPU与动态分配GPU memory设置
Feb 03 #Python
TensorFlow内存管理bfc算法实例
Feb 03 #Python
python numpy数组中的复制知识解析
Feb 03 #Python
opencv python Canny边缘提取实现过程解析
Feb 03 #Python
Pycharm debug调试时带参数过程解析
Feb 03 #Python
Python使用enumerate获取迭代元素下标
Feb 03 #Python
You might like
用PHP中的 == 运算符进行字符串比较
2006/11/26 PHP
php创建、获取cookie及基础要点分析
2015/01/26 PHP
PHP运用foreach神奇的转换数组(实例讲解)
2018/02/01 PHP
Yii框架参数配置文件params用法实例分析
2019/09/11 PHP
Gambit vs CL BO3 第二场 2.13
2021/03/10 DOTA
ASP.NET jQuery 实例18 通过使用jQuery validation插件校验DropDownList
2012/02/03 Javascript
利用JS进行图片的切换即特效展示图片
2013/12/03 Javascript
js常用系统函数用法实例分析
2015/01/12 Javascript
jquery+css3实现网页背景花瓣随机飘落特效
2015/08/17 Javascript
基于jQuery+PHP+Mysql实现在线拍照和在线浏览照片
2015/09/06 Javascript
javascript实现tab切换的四种方法
2015/11/05 Javascript
jQuery中hover与mouseover和mouseout的区别分析
2015/12/24 Javascript
TypeScript Type Innference(类型判断)
2016/03/10 Javascript
省市选择的简单实现(基于zepto.js)
2016/06/21 Javascript
jquery——九宫格大转盘抽奖实例
2017/01/16 Javascript
seajs中最常用的7个功能、配置示例
2017/10/10 Javascript
js调用设备摄像头的方法
2018/07/19 Javascript
vue-cli 默认路由再子路由选中下的选中状态问题及解决代码
2018/09/06 Javascript
angularjs模态框的使用代码实例
2019/12/20 Javascript
Vue使用路由钩子拦截器beforeEach和afterEach监听路由
2020/11/16 Javascript
Python常用模块logging——日志输出功能(示例代码)
2019/11/20 Python
Python turtle库的画笔控制说明
2020/06/28 Python
vscode调试django项目的方法
2020/08/06 Python
Python爬取股票信息,并可视化数据的示例
2020/09/26 Python
一些常用的HTML5模式(pattern) 总结
2015/07/14 HTML / CSS
德国自行车商店:Tretwerk
2019/06/21 全球购物
党员自我评价分享
2013/12/13 职场文书
土建资料员岗位职责
2014/01/04 职场文书
中学家长会邀请函
2014/02/03 职场文书
2014年教师节演讲稿
2014/09/03 职场文书
学校领导班子对照检查材料
2014/09/24 职场文书
2015年教师党员自我评价材料
2015/03/04 职场文书
2015年全民国防教育日活动总结
2015/03/23 职场文书
2015年学生管理工作总结
2015/05/26 职场文书
垂直极限观后感
2015/06/08 职场文书
CSS3点击按钮圆形进度打钩效果的实现代码
2021/03/30 HTML / CSS