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 相关文章推荐
PyMongo安装使用笔记
Apr 27 Python
Saltstack快速入门简单汇总
Mar 01 Python
import的本质解析
Oct 30 Python
numpy基础教程之np.linalg
Feb 12 Python
python中yield的用法详解——最简单,最清晰的解释
Apr 04 Python
python [:3] 实现提取数组中的数
Nov 27 Python
使用Pytorch来拟合函数方式
Jan 14 Python
pytorch实现线性拟合方式
Jan 15 Python
Python使用qrcode二维码库生成二维码方法详解
Feb 17 Python
解决paramiko执行命令超时的问题
Apr 16 Python
python 写一个文件分发小程序
Dec 05 Python
Python jieba库分词模式实例用法
Jan 13 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
第1次亲密接触PHP5(2)
2006/10/09 PHP
PHP中鲜为人知的10个函数
2014/02/28 PHP
PHP接口并发测试的方法(推荐)
2016/12/15 PHP
删除PHP数组中头部、尾部、任意元素的实现代码
2017/04/10 PHP
通过源码解析Laravel的依赖注入
2018/01/22 PHP
js资料toString 方法
2007/03/13 Javascript
JavaScript replace(rgExp,fn)正则替换的用法
2010/03/04 Javascript
jquery 实现表单验证功能代码(简洁)
2012/07/03 Javascript
jQuery滚动加载图片效果的实现
2013/03/06 Javascript
javascript实现跨域的方法汇总
2015/06/25 Javascript
一个简单的JavaScript Map实例(分享)
2016/08/03 Javascript
JS清除字符串中重复值的实现方法
2016/08/03 Javascript
基于javascript实现按圆形排列DIV元素(一)
2016/12/02 Javascript
vuejs2.0运用原生js实现简单的拖拽元素功能示例
2017/02/24 Javascript
vue.js实现备忘录功能的方法
2017/07/10 Javascript
node 利用进程通信实现Cluster共享内存
2017/10/27 Javascript
javascript性能优化之分时函数的介绍
2018/03/28 Javascript
微信小程序使用component自定义toast弹窗效果
2018/11/27 Javascript
VSCode使用之Vue工程配置eslint
2019/04/30 Javascript
微信公众号服务器验证Token步骤图解
2019/12/30 Javascript
JavaScript Window浏览器对象模型原理解析
2020/05/30 Javascript
vue表单验证之禁止input输入框输入空格
2020/12/03 Vue.js
Vue多选列表组件深入详解
2021/03/02 Vue.js
Python中的闭包实例详解
2014/08/29 Python
浅析python参数的知识点
2018/12/10 Python
使用tqdm显示Python代码执行进度功能
2019/12/08 Python
python 定义类时,实现内部方法的互相调用
2019/12/25 Python
浅谈Pycharm最有必要改的几个默认设置项
2020/02/14 Python
css3的transition效果和transfor效果示例介绍
2013/10/30 HTML / CSS
Ibood荷兰:互联网每日最佳在线优惠
2019/02/28 全球购物
JAVA代码查错题
2014/10/10 面试题
一些Solaris面试题
2013/03/22 面试题
建筑安全员岗位职责
2014/03/13 职场文书
党员干部承诺书
2014/03/25 职场文书
道歉的话怎么说
2015/05/12 职场文书
助学金申请书该怎么写?
2019/07/16 职场文书