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 相关文章推荐
详解Python3中yield生成器的用法
Aug 20 Python
用python 批量更改图像尺寸到统一大小的方法
Mar 31 Python
python搭建服务器实现两个Android客户端间收发消息
Apr 12 Python
python 利用栈和队列模拟递归的过程
May 29 Python
Python 计算任意两向量之间的夹角方法
Jul 05 Python
解决Python对齐文本字符串问题
Aug 28 Python
Django使用消息提示简单的弹出个对话框实例
Nov 15 Python
Python turtle库绘制菱形的3种方式小结
Nov 23 Python
记一次pyinstaller打包pygame项目为exe的过程(带图片)
Mar 02 Python
python定义类的简单用法
Jul 24 Python
python中zip()函数遍历多个列表方法
Feb 18 Python
Python词云的正确实现方法实例
May 08 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
轻松入门: 煮好咖啡的七个诀窍
2021/03/03 冲泡冲煮
php时间不正确的解决方法
2008/04/09 PHP
简单示例AJAX结合PHP代码实现登录效果代码
2008/07/25 PHP
typecho插件编写教程(五):核心代码
2015/05/28 PHP
使两个iframe的高度与内容自适应,且相等
2006/11/20 Javascript
js实现GridView单选效果自动设置交替行、选中行、鼠标移动行背景色
2010/05/27 Javascript
初窥JQuery(二) 事件机制(1)
2010/11/25 Javascript
基于jQuery的获取标签名的代码
2012/07/16 Javascript
javascrip关于继承的小例子
2013/05/10 Javascript
JavaScript闭包实例讲解
2014/04/22 Javascript
jquery uploadify 在FF下无效的解决办法
2014/09/26 Javascript
jQuery插件zoom实现图片全屏放大弹出层特效
2015/04/15 Javascript
Bootstrap每天必学之导航
2015/11/26 Javascript
深入浅析AngularJS和DataModel
2016/02/16 Javascript
js 获取本地文件及目录的方法(推荐)
2016/11/10 Javascript
Bootstrap table 定制提示语的加载过程
2017/02/20 Javascript
js中的闭包实例展示
2018/11/01 Javascript
详解如何使用node.js的开发框架express创建一个web应用
2018/12/20 Javascript
node.js Promise对象的使用方法实例分析
2019/12/26 Javascript
[38:31]完美世界DOTA2联赛PWL S3 Magma vs GXR 第一场 12.13
2020/12/17 DOTA
python进阶教程之循环相关函数range、enumerate、zip
2014/08/30 Python
python网络编程之读取网站根目录实例
2014/09/30 Python
Python中文分词实现方法(安装pymmseg)
2016/06/14 Python
对python中Librosa的mfcc步骤详解
2019/01/09 Python
Python使用Pandas对csv文件进行数据处理的方法
2019/08/01 Python
python实现猜数游戏
2020/03/27 Python
pytorch 查看cuda 版本方式
2020/06/23 Python
彪马土耳其官网:PUMA土耳其
2019/07/14 全球购物
医院学雷锋活动策划方案
2014/02/15 职场文书
公证委托书格式
2014/09/13 职场文书
作风整顿剖析材料
2014/09/30 职场文书
写给父母的感谢信
2015/01/22 职场文书
故意杀人罪辩护词
2015/05/21 职场文书
Java 实战项目之家居购物商城系统详解流程
2021/11/11 Java/Android
python 判断文件或文件夹是否存在
2022/03/18 Python
Python内置数据类型中的集合详解
2022/03/18 Python