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使用arp欺骗伪造网关的方法
Apr 24 Python
详解Python中的Numpy、SciPy、MatPlotLib安装与配置
Nov 17 Python
python,Django实现的淘宝客登录功能示例
Jun 12 Python
Python Web框架之Django框架Form组件用法详解
Aug 16 Python
python列表推导式操作解析
Nov 26 Python
jupyter修改文件名方式(TensorFlow)
Apr 21 Python
PyQt5实现登录页面
May 30 Python
python 基于卡方值分箱算法的实现示例
Jul 17 Python
Pandas的Apply函数具体使用
Jul 21 Python
python字典key不能是可以是啥类型
Aug 04 Python
K近邻法(KNN)相关知识总结以及如何用python实现
Jan 28 Python
Python基本的内置数据类型及使用方法
Apr 13 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
frename PHP 灵活文件命名函数 frename
2009/09/09 PHP
php实现以只读方式打开文件的方法
2015/03/16 PHP
php中file_exists函数使用详解
2015/05/08 PHP
完美解决在ThinkPHP控制器中命名空间的问题
2017/05/05 PHP
PHP封装XML和JSON格式数据接口操作示例
2019/03/06 PHP
Zero Clipboard js+swf实现的复制功能使用方法
2010/03/07 Javascript
javscript对象原型的一些看法
2010/09/19 Javascript
Jquery读取URL参数小例子
2013/08/30 Javascript
jquery插件jquery倒计时插件分享
2013/12/27 Javascript
javascript event在FF和IE的兼容传参心得(绝对好用)
2014/07/10 Javascript
Javascript实现计算个人所得税
2015/05/10 Javascript
JS提交form表单实例分析
2015/12/10 Javascript
Node.js测试中的Mock文件系统详解
2016/11/21 Javascript
JS实现的RGB网页颜色在线取色器完整实例
2016/12/21 Javascript
解决Webpack 热部署检测不到文件变化的问题
2018/02/22 Javascript
JavaScript剩余操作符Rest Operator详解
2019/07/20 Javascript
[13:21]DOTA2国际邀请赛采访专栏:RSnake战队国士无双,Fnatic.Fly
2013/08/06 DOTA
Python中的is和id用法分析
2015/01/26 Python
Python中操作MySQL入门实例
2015/02/08 Python
Python中IPYTHON入门实例
2015/05/11 Python
Python实现telnet服务器的方法
2015/07/10 Python
Python正则表达式教程之三:贪婪/非贪婪特性
2017/03/02 Python
python实现音乐下载器
2018/04/15 Python
在pycharm中使用git版本管理以及同步github的方法
2019/01/16 Python
Python制作exe文件简单流程
2019/01/24 Python
django认证系统 Authentication使用详解
2019/07/22 Python
Pandas透视表(pivot_table)详解
2019/07/22 Python
python3中替换python2中cmp函数的实现
2019/08/20 Python
python3文件复制、延迟文件复制任务的实现方法
2019/09/02 Python
解决pycharm 安装numpy失败的问题
2019/12/05 Python
python安装后的目录在哪里
2020/06/21 Python
世界上最大的隐形眼镜商店:1-800 Contacts
2018/11/03 全球购物
VC++笔试题
2014/10/13 面试题
2014年医生工作总结
2014/11/21 职场文书
解决Goland 同一个package中函数互相调用的问题
2021/05/06 Golang
Python进度条的使用
2021/05/17 Python