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函数学习笔记
Oct 07 Python
centos下更新Python版本的步骤
Feb 12 Python
python实现百度关键词排名查询
Mar 30 Python
Python多线程编程(一):threading模块综述
Apr 05 Python
Python的语言类型(详解)
Jun 24 Python
TensorFlow如何实现反向传播
Feb 06 Python
python matplotlib 在指定的两个点之间连线方法
May 25 Python
APIStar:一个专为Python3设计的API框架
Sep 26 Python
解决Python 命令行执行脚本时,提示导入的包找不到的问题
Jan 19 Python
Python检测数据类型的方法总结
May 20 Python
jupyter notebook实现显示行号
Apr 13 Python
Python类及获取对象属性方法解析
Jun 15 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 mysql事务问题实例分析
2016/01/18 PHP
PHP使用星号替代用户名手机和邮箱的实现代码
2018/02/07 PHP
php实现生成PDF文件的方法示例【基于FPDF类库】
2018/07/21 PHP
javascript显示隐藏层比较不错的方法分析
2008/09/30 Javascript
js对象的比较
2011/02/26 Javascript
javascript parseInt() 函数的进制转换注意细节
2013/01/08 Javascript
asm.js使用示例代码
2013/11/28 Javascript
文本框文本自动补全效果示例分享
2014/01/19 Javascript
禁止页面刷新让F5快捷键及右键都无效
2014/01/22 Javascript
Jquery EasyUI中弹出确认对话框以及加载效果示例代码
2014/02/13 Javascript
详解JavaScript中this关键字的用法
2016/05/26 Javascript
vue中如何引入jQuery和Bootstrap
2017/04/10 jQuery
原生js中ajax访问的实例详解
2017/09/19 Javascript
AngularJS实现注册表单验证功能
2017/10/16 Javascript
详解如何用VUE写一个多用模态框组件模版
2018/09/27 Javascript
vue表单验证你真的会了吗?vue表单验证(form)validate
2019/04/07 Javascript
JavaScript arguments.callee作用及替换方案详解
2020/09/02 Javascript
[59:44]2018DOTA2亚洲邀请赛 3.31 小组赛 B组 paiN vs iG
2018/03/31 DOTA
Python写的Discuz7.2版faq.php注入漏洞工具
2014/08/06 Python
Python and、or以及and-or语法总结
2015/04/14 Python
Python中easy_install 和 pip 的安装及使用
2017/06/05 Python
Python统计单词出现的次数
2018/04/04 Python
python学习基础之循环import及import过程
2018/04/22 Python
python定位xpath 节点位置的方法
2019/08/27 Python
python下载库的步骤方法
2019/10/12 Python
Python基本类型的连接组合和互相转换方式(13种)
2019/12/16 Python
pytorch 中pad函数toch.nn.functional.pad()的用法
2020/01/08 Python
Jupyter notebook运行Spark+Scala教程
2020/04/10 Python
Pyecharts 中Geo函数常用参数的用法说明
2021/02/01 Python
Java 中访问数据库的步骤?Statement 和PreparedStatement 之间的区别?
2012/06/05 面试题
医学生个人求职信范文
2014/02/07 职场文书
2014年平安夜寄语
2014/12/08 职场文书
小学中等生评语
2014/12/29 职场文书
电影焦裕禄观后感
2015/06/09 职场文书
2015年社区国庆节活动总结
2015/07/30 职场文书
MySQL的索引你了解吗
2022/03/13 MySQL