Keras设定GPU使用内存大小方式(Tensorflow backend)


Posted in Python onMay 22, 2020

通过设置Keras的Tensorflow后端的全局变量达到。

import os
import tensorflow as tf
import keras.backend.tensorflow_backend as KTF
 
def get_session(gpu_fraction=0.3):
 '''Assume that you have 6GB of GPU memory and want to allocate ~2GB'''
 
 num_threads = os.environ.get('OMP_NUM_THREADS')
 gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_fraction)
 
 if num_threads:
  return tf.Session(config=tf.ConfigProto(
   gpu_options=gpu_options, intra_op_parallelism_threads=num_threads))
 else:
  return tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

使用过程中显示的设置session:

import keras.backend.tensorflow_backend as KTF
KTF.set_session(get_session())

补充知识:限制tensorflow的运行内存 (keras.backend.tensorflow)

我就废话不多说了,大家还是直接看代码吧!

import tensorflow as tf
from keras.backend.tensorflow_backend import set_session

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5 #half of the memory
set_session(tf.Session(config=config))

以上这篇Keras设定GPU使用内存大小方式(Tensorflow backend)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python max内置函数详细介绍
Nov 17 Python
python+opencv实现的简单人脸识别代码示例
Nov 14 Python
利用Python在一个文件的头部插入数据的实例
May 02 Python
Python  unittest单元测试框架的使用
Sep 08 Python
Pandas DataFrame 取一行数据会得到Series的方法
Nov 10 Python
将Python字符串生成PDF的实例代码详解
May 17 Python
在pycharm下设置自己的个性模版方法
Jul 15 Python
Python matplotlib绘制饼状图功能示例
Sep 10 Python
Python 实现黑客帝国中的字符雨的示例代码
Feb 20 Python
解决 jupyter notebook 回车换两行问题
Apr 15 Python
python判断变量是否为列表的方法
Sep 17 Python
SpringBoot首页设置解析(推荐)
Feb 11 Python
tensorflow使用L2 regularization正则化修正overfitting过拟合方式
May 22 #Python
Softmax函数原理及Python实现过程解析
May 22 #Python
Python接口测试文件上传实例解析
May 22 #Python
计算Python Numpy向量之间的欧氏距离实例
May 22 #Python
python numpy矩阵信息说明,shape,size,dtype
May 22 #Python
python查看矩阵的行列号以及维数方式
May 22 #Python
30行Python代码实现高分辨率图像导航的方法
May 22 #Python
You might like
PHP学习笔记之二
2011/01/17 PHP
PHP函数学习之PHP函数点评
2012/07/05 PHP
php生成酷炫的四个字符验证码
2016/04/22 PHP
深入理解PHP原理之执行周期分析
2016/06/01 PHP
ucenter中词语过滤原理分析
2016/07/13 PHP
PHP依赖注入原理与用法分析
2018/08/21 PHP
PHP中Session ID的实现原理实例分析
2019/08/17 PHP
经常用的图片在容器中的水平垂直居中实例
2007/06/10 Javascript
javascript操作cookie的文章(设置,删除cookies)
2010/04/01 Javascript
JS打开新窗口的2种方式
2013/04/18 Javascript
javaScript如何生成xmlhttp
2013/12/16 Javascript
JavaScript将当前时间转换成UTC标准时间的方法
2015/04/06 Javascript
javascript日期操作详解(脚本之家整理)
2015/09/05 Javascript
基于MVC4+EasyUI的Web开发框架形成之旅之界面控件的使用
2015/12/16 Javascript
jQuery中的select操作详解
2016/11/29 Javascript
vue-cli 3.x 修改dist路径的方法
2018/09/19 Javascript
详解在create-react-app使用less与antd按需加载
2018/12/06 Javascript
js中事件对象和事件委托的介绍
2019/01/21 Javascript
Vue组件间通信 Vuex的用法解析
2019/08/05 Javascript
微信小程序实现左滑删除效果
2020/11/18 Javascript
Vue实现多页签组件
2021/01/14 Vue.js
python生成excel的实例代码
2017/11/08 Python
python 拷贝特定后缀名文件,并保留原始目录结构的实例
2018/04/27 Python
Expected conditions模块使用方法汇总代码解析
2020/08/13 Python
详解基于Scrapy的IP代理池搭建
2020/09/29 Python
使用django自带的user做外键的方法
2020/11/30 Python
html5简介_动力节点Java学院整理
2017/07/07 HTML / CSS
浅谈Html5移动端ios/Android兼容性总结
2018/06/01 HTML / CSS
日本钓鱼渔具和户外用品网上商店:naturum
2016/08/07 全球购物
中文专业毕业生自荐信
2013/10/28 职场文书
产品质量承诺范本
2014/03/31 职场文书
实习生评语
2014/04/26 职场文书
食品流通安全承诺书
2014/05/22 职场文书
银行授权委托书格式
2014/10/10 职场文书
运动会1000米加油稿
2015/07/21 职场文书
微信小程序结合ThinkPHP5授权登陆后获取手机号
2021/11/23 PHP