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采用getopt解析命令行输入参数实例
Sep 30 Python
python脚本替换指定行实现步骤
Jul 11 Python
python利用有道翻译实现"语言翻译器"的功能实例
Nov 14 Python
python3.7.0的安装步骤
Aug 27 Python
Django框架组成结构、基本概念与文件功能分析
Jul 30 Python
Python 一键获取百度网盘提取码的方法
Aug 01 Python
Python3加密解密库Crypto的RSA加解密和签名/验签实现方法实例
Feb 11 Python
简单的命令查看安装的python版本号
Aug 28 Python
Python3实现英文字母转换哥特式字体实例代码
Sep 01 Python
浅析PyCharm 的初始设置(知道)
Oct 12 Python
Python 微信公众号文章爬取的示例代码
Nov 30 Python
总结python多进程multiprocessing的相关知识
Jun 29 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
2019十大人气国漫
2020/03/13 国漫
PHP令牌 Token改进版
2008/07/18 PHP
php 安全过滤函数代码
2011/05/07 PHP
PHP内核探索之解释器的执行过程
2015/12/22 PHP
PHP内存缓存功能memcached示例
2016/10/19 PHP
php查找字符串中第一个非0的位置截取
2017/02/27 PHP
showModalDialog 和 showModelessDialog
2007/01/22 Javascript
收集的10个免费的jQuery相册
2011/02/26 Javascript
判断JS对象是否拥有某种属性的两种方式
2013/12/02 Javascript
用jquery模仿的a的title属性的例子
2014/10/22 Javascript
js实现百度联盟中一款不错的图片切换效果完整实例
2015/03/04 Javascript
JavaScript的React Web库的理念剖析及基础上手指南
2016/05/10 Javascript
JS调用Android、Ios原生控件
2017/01/06 Javascript
MUI 解决动态列表页图片懒加载再次加载不成功的bug问题
2017/04/13 Javascript
详解Vue双向数据绑定原理解析
2017/09/11 Javascript
Vue2.0 实现单选互斥的方法
2018/04/13 Javascript
详解Angular6.0使用路由步骤(共7步)
2018/06/29 Javascript
JS左右无缝轮播功能完整实例
2019/05/16 Javascript
关于vue2强制刷新,解决页面不会重新渲染的问题
2019/10/29 Javascript
ant design实现圈选功能
2019/12/17 Javascript
vue子组件改变父组件传递的prop值通过sync实现数据双向绑定(DEMO)
2020/02/01 Javascript
python模拟登录百度代码分享(获取百度贴吧等级)
2013/12/27 Python
python遍历类中所有成员的方法
2015/03/18 Python
《与孩子一起学编程》python自测题
2018/05/27 Python
Python matplotlib的使用并自定义colormap的方法
2018/12/13 Python
python实现批量nii文件转换为png图像
2019/07/18 Python
Python 使用指定的网卡发送HTTP请求的实例
2019/08/21 Python
python如何删除列为空的行
2020/07/17 Python
利于python脚本编写可视化nmap和masscan的方法
2020/12/29 Python
介绍一下linux的文件权限
2012/02/15 面试题
EJB实例的生命周期
2016/10/28 面试题
旅游市场营销方案
2014/03/09 职场文书
党员批评与自我批评思想汇报
2014/10/08 职场文书
群众路线教育实践活动总结
2014/10/30 职场文书
2015年创先争优工作总结
2015/05/23 职场文书
导游词之无锡华莱坞
2019/12/02 职场文书