解决Keras使用GPU资源耗尽的问题
更新时间：2020年06月22日 10:57:20 作者：wp1988
这篇文章主要介绍了解决Keras使用GPU资源耗尽的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
我们在使用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资源耗尽的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
大家好，本篇文章主要讲的是Android申请相机权限和读写权限实例，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下2022-02-02
今天小编就为大家分享一篇python判断字符串或者集合是否为空的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
这篇文章主要介绍了使用Keras加载含有自定义层或函数的模型操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
这篇文章主要介绍了用python拟合等角螺线的实现示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-12-12
这篇文章主要介绍了Python中遍历字典过程中更改元素导致错误的解决方法,针对增删元素后出现dictionary changed size during iteration的异常解决做出讨论和解决,需要的朋友可以参考下2016-05-05
这篇文章主要详细介绍了Python列表reverse()函数使用方法，文章通过代码示例讲解的非常详细，对我们的学习或工作有一定的帮助,需要的朋友可以参考下2023-07-07
今天小编就为大家分享一篇解决Python3 被PHP程序调用执行返回乱码的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-02-02
Django shell调试models输出的SQL语句方法
今天小编就为大家分享一篇Django shell调试models输出的SQL语句方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
这篇文章主要介绍了Python引入多个模块及包的概念过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-09-09
pygame是用来开发游戏的一套基于SDL的模板,它可以是python创建完全界面化的游戏和多媒体程序,而且它基本上可以在任何系统上运行,这篇文章主要给大家介绍了基于python pygame实现的兔子吃月饼小游戏的相关资料,需要的朋友可以参考下2021-09-09
最新评论