这个屌丝很懒，什么也没留下！

Keras按需分配使用GPU_keras 加为不同模型分配显卡占比

作者：代码秘术家 | 2024-02-01 12:22:56

keras 加为不同模型分配显卡占比

在使用keras作为神经网络的架构时，如果不定义按需分配使用GPU，则会占满GPU,可以通过以下定义按需分配：

import tensorflow as tf
import keras.backend.tensorflow_backend as K

config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)
K.set_session(sess)
1
2
3
4
5
6
7

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/article/detail/53765