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脚本
Sep 01 Python
python中的__slots__使用示例
Feb 26 Python
python中的内置函数max()和min()及mas()函数的高级用法
Mar 29 Python
python中的decorator的作用详解
Jul 26 Python
Python subprocess库的使用详解
Oct 26 Python
Mac下Anaconda的安装和使用教程
Nov 29 Python
对python的bytes类型数据split分割切片方法
Dec 04 Python
Python tkinter实现图片标注功能(完整代码)
Dec 08 Python
Python性能分析工具py-spy原理用法解析
Jul 27 Python
python实现高效的遗传算法
Apr 07 Python
python代码实现备忘录案例讲解
Jul 26 Python
python语言中pandas字符串分割str.split()函数
Aug 05 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获取文件夹路径内的图片以及分页显示示例
2014/03/11 PHP
CentOS6.5 编译安装lnmp环境
2014/12/21 PHP
php格式化电话号码的方法
2015/04/24 PHP
redirect_uri参数错误的解决方法(必看)
2017/02/16 PHP
JS宝典学习笔记(下)
2007/01/10 Javascript
javascript,jquery闭包概念分析
2010/06/19 Javascript
JavaScript 放大镜 放大倍率和视窗尺寸
2011/05/09 Javascript
jquery UI Datepicker时间控件的使用方法(加强版)
2015/11/07 Javascript
AngularJS基础 ng-keyup 指令简单示例
2016/08/02 Javascript
使用bat打开多个cmd窗口执行gulp、node
2017/02/17 Javascript
利用forever和pm2部署node.js项目过程
2017/05/10 Javascript
JavaScript函数、闭包、原型、面向对象学习笔记
2018/09/06 Javascript
wx-charts 微信小程序图表插件的具体使用
2019/08/18 Javascript
JavaScript中0、空字符串、'0'是true还是false的知识点分享
2019/09/16 Javascript
使用vue实现HTML页面生成图片的方法
2020/03/12 Javascript
使用Python脚本生成随机IP的简单方法
2015/07/30 Python
Python实现中一次读取多个值的方法
2018/04/22 Python
使用python爬取B站千万级数据
2018/06/08 Python
Python内置加密模块用法解析
2019/11/25 Python
python装饰器的特性原理详解
2019/12/25 Python
opencv python Canny边缘提取实现过程解析
2020/02/03 Python
Python直接赋值及深浅拷贝原理详解
2020/09/05 Python
python 实现aes256加密
2020/11/27 Python
工程业务员工作职责
2013/12/07 职场文书
2014学雷锋活动心得体会
2014/03/10 职场文书
小学家长评语大全
2014/04/16 职场文书
写给导师的自荐信
2015/03/06 职场文书
2016年会领导致辞稿
2015/07/29 职场文书
初级职称评定工作总结
2015/08/13 职场文书
2016七夕情人节广告语
2016/01/28 职场文书
新员工入职感言范文!
2019/07/04 职场文书
Python编程源码报错解决方法总结经验分享
2021/10/05 Python
利用JavaScript写一个简单计算器
2021/11/27 Javascript
python数字图像处理:图像简单滤波
2022/06/28 Python
PostgreSQL逻辑复制解密原理解析
2022/09/23 PostgreSQL