Keras按需分配使用GPU
Keras按需分配使用GPU
在使用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)
原创
2020-12-18 10:01:26 ·
343 阅读 ·
0 评论