Keras设定GPU使用内存大小方式(Tensorflow backend)
更新时间：2020年05月22日 17:14:10 作者：mjiansun
这篇文章主要介绍了Keras设定GPU使用内存大小方式(Tensorflow backend)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
通过设置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)就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇解决python3 Pycharm上连接数据库时报错的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
这篇文章主要介绍了python中正则表达式的使用详解,另附上一些常用的Python正则例子，有需要的朋友可以参考下2014-10-10
今天小编就为大家分享一篇Python 将Matrix、Dict保存到文件的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-10-10
本篇文章了tensorflow训练自己的数据集实现CNN图像分类，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-02-02
Python中ModuleNotFoundError错误的问题解决
本文主要介绍了Python中ModuleNotFoundError错误的问题解决，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-02-02
在Python编程语言中，变量是用于存储数据值的标识符，它们可以用来引用数据值，而不是直接使用值本身，本文将详细介绍python中的变量，感兴趣的同学可以参考阅读2023-05-05
这篇文章主要为大家详细介绍了python书籍信息爬虫示例，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-03-03
这篇文章主要介绍了解决Django生产环境无法加载静态文件问题的解决，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2019-04-04
python中format函数用于字符串的格式化，这篇文章主要介绍了python中强大的format函数,需要的朋友可以参考下2018-12-12
这篇文章主要为大家介绍了Python UI自动化测试Web frame及多窗口切换，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-11-11
最新评论