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实现忽略大小写对字符串列表排序的方法
Sep 25 Python
详解Python3 中hasattr()、getattr()、setattr()、delattr()函数及示例代码数
Apr 18 Python
python3利用tcp实现文件夹远程传输
Jul 28 Python
Python数据可视化库seaborn的使用总结
Jan 15 Python
python实现点击按钮修改数据的方法
Jul 17 Python
Python使用matplotlib 画矩形的三种方式分析
Oct 31 Python
Python上下文管理器用法及实例解析
Nov 11 Python
Python计算不规则图形面积算法实现解析
Nov 22 Python
Python openpyxl 插入折线图实例
Apr 17 Python
python 实现波浪滤镜特效
Dec 02 Python
Python中的流程控制详解
Feb 18 Python
Python利用机器学习算法实现垃圾邮件的识别
Jun 28 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图片的裁剪与缩放生成符合需求的缩略图
2013/01/11 PHP
PHP利用超级全局变量$_GET来接收表单数据的实例
2016/11/05 PHP
Django中通过定时任务触发页面静态化的处理方式
2018/08/29 PHP
PHP设计模式(八)装饰器模式Decorator实例详解【结构型】
2020/05/02 PHP
jQuery实现随意改变div任意属性的名称和值(部分原生js实现)
2013/05/28 Javascript
JavaScript中圆括号()和方括号[]的特殊用法疑问解答
2013/08/06 Javascript
jquery提交form表单时禁止重复提交的方法
2014/02/13 Javascript
jquery等待效果示例
2014/05/01 Javascript
JS中prototype的用法实例分析
2015/03/19 Javascript
jquery实现两边飘浮可关闭的对联广告
2015/11/27 Javascript
Jquery与Bootstrap实现后台管理页面增删改查功能示例
2017/01/22 Javascript
jQuery接受后台传递的List的实例详解
2017/08/02 jQuery
加载 vue 远程代码的组件实例详解
2017/11/20 Javascript
jquery树形插件zTree高级使用详解
2019/08/16 jQuery
es6函数之尾递归用法实例分析
2020/04/25 Javascript
使用PYTHON接收多播数据的代码
2012/03/01 Python
浅谈Python 中整型对象的存储问题
2016/05/16 Python
Python实现基本线性数据结构
2016/08/22 Python
django站点管理详解
2017/12/12 Python
Python网络编程使用select实现socket全双工异步通信功能示例
2018/04/09 Python
pandas Dataframe行列读取的实例
2018/06/08 Python
基于数据归一化以及Python实现方式
2018/07/11 Python
Scrapy使用的基本流程与实例讲解
2018/10/21 Python
docker django无法访问redis容器的解决方法
2019/08/21 Python
Python自动重新加载模块详解(autoreload module)
2020/04/01 Python
Django启动时找不到mysqlclient问题解决方案
2020/11/11 Python
Python爬虫之Selenium下拉框处理的实现
2020/12/04 Python
python将YUV420P文件转PNG图片格式的两种方法
2021/01/22 Python
秋季开学典礼主持词
2014/03/19 职场文书
党员对十八届四中全会的期盼思想汇报范文
2014/10/17 职场文书
2014年团委工作总结
2014/11/13 职场文书
企业年会祝酒词
2015/08/11 职场文书
关于职业道德的心得体会
2016/01/18 职场文书
python实现对doc、txt、xls等文档的读写操作
2022/04/02 Python
MySQL三种方式实现递归查询
2022/04/18 MySQL
php解析非标准json、非规范json的方式实例
2022/05/10 PHP