解决Keras使用GPU资源耗尽的问题


Posted in Python onJune 22, 2020

我们在使用GPU资源进行训练的时候,可能会发生资源耗尽的情况,那么在在这种情况,我们需要对GPU的资源进行合理的安排,具体使用办法如下:

框架:Tensorflow和Keras

方法

import tensorflow as tf
import keras.backend.tensorflow_backend as KTF
config = tf.ConfigProto()
config.gpu_options.allow_growth=True #不全部占满显存, 按需分配
sess = tf.Session(config=config)
KTF.set_session(sess) # 设置session

通过这种方法,就能合理的使用GPU资源了。

至少到目前位置,我自己从程序没出现资源耗尽的情况,当然,对于batchsize的设置,一定要设置在合理的范围,所谓合理,大家自己体会。

补充知识:keras使用GPU的一些坑

keras安装

conda install tensorflow-gpu

pip install keras

注意:不要使用conda安装keras,会检测依赖,默认安装上tensorflow的CPU版本。

keras运行报错,tensorflow/stream_executor/cuda/cuda_dnn.cc:329 Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

解决方法:

增加如下代码:

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.5)
config = tf.ConfigProto(gpu_options=gpu_options)
config.gpu_options.allow_growth = True
session = tf.Session(config=config)

以上这篇解决Keras使用GPU资源耗尽的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python 网络编程常用代码段
Aug 28 Python
Python使用win32com实现的模拟浏览器功能示例
Jul 13 Python
python 每天如何定时启动爬虫任务(实现方法分享)
May 21 Python
python-opencv颜色提取分割方法
Dec 08 Python
Python中使用遍历在列表中添加字典遇到的坑
Feb 27 Python
50行Python代码获取高考志愿信息的实现方法
Jul 23 Python
在Python中利用pickle保存变量的实例
Dec 30 Python
pycharm 激活码及使用方式的详细教程
May 12 Python
Python smtp邮件发送模块用法教程
Jun 15 Python
python绘制分布折线图的示例
Sep 24 Python
python引入其他文件夹下的py文件具体方法
May 23 Python
Python列表的索引与切片
Apr 07 Python
Keras - GPU ID 和显存占用设定步骤
Jun 22 #Python
Python 基于jwt实现认证机制流程解析
Jun 22 #Python
python中format函数如何使用
Jun 22 #Python
Tensorflow与Keras自适应使用显存方式
Jun 22 #Python
python数据类型强制转换实例详解
Jun 22 #Python
keras 指定程序在某块卡上训练实例
Jun 22 #Python
python Socket网络编程实现C/S模式和P2P
Jun 22 #Python
You might like
使用Linux五年积累的一些经验技巧
2013/06/20 PHP
php实现在限定区域里自动调整字体大小的类实例
2015/04/02 PHP
PHP接口继承及接口多继承原理与实现方法详解
2017/10/18 PHP
PHP自定义序列化接口Serializable用法分析
2017/12/29 PHP
js 跨域和ajax 跨域问题小结
2009/07/01 Javascript
仅Firefox中链接A无法实现模拟点击以触发其默认行为
2011/07/31 Javascript
JS如何将数字类型转化为没3个一个逗号的金钱格式
2014/01/27 Javascript
JavaScript通过事件代理高亮显示表格行的方法
2015/05/27 Javascript
深入理解JavaScript编程中的同步与异步机制
2015/06/24 Javascript
JavaScript中Boolean对象的属性解析
2015/10/21 Javascript
jQuery仿京东商城楼梯式导航定位菜单
2016/07/25 Javascript
判断js的Array和Object的实现方法
2016/08/29 Javascript
[原创]JS基于FileSaver.js插件实现文件保存功能示例
2016/12/08 Javascript
vuejs响应用户事件(如点击事件)
2017/03/14 Javascript
vue.js 嵌套循环、if判断、动态删除的实例
2018/03/07 Javascript
jQuery实现轮播图及其原理详解
2020/04/12 jQuery
js实现漂亮的星空背景
2019/11/01 Javascript
Vue 中使用lodash对事件进行防抖和节流操作
2020/07/26 Javascript
解决Can't find variable: SockJS vue项目的问题
2020/09/22 Javascript
Python中bisect的用法
2014/09/23 Python
C#返回当前系统所有可用驱动器符号的方法
2015/04/18 Python
python 数据的清理行为实例详解
2017/07/12 Python
python中列表的切片与修改知识点总结
2019/07/23 Python
python 回溯法模板详解
2020/02/26 Python
详解如何使用Pytest进行自动化测试
2021/01/14 Python
非常漂亮的CSS3百叶窗焦点图动画
2016/02/24 HTML / CSS
中国跨境海淘网站:考拉海购
2016/08/01 全球购物
自荐书模板
2013/12/19 职场文书
《蚕姑娘》教学反思
2014/04/15 职场文书
工会工作先进事迹
2014/08/18 职场文书
大学生党员批评与自我批评范文
2014/10/14 职场文书
2015年街道除四害工作总结
2015/05/15 职场文书
追讨欠款律师函
2015/06/24 职场文书
公文写作:教你写“建议书”
2019/05/07 职场文书
Python爬虫:从m3u8文件里提取小视频的正确操作
2021/05/14 Python
Java基础-封装和继承
2021/07/02 Java/Android