環(huán)境說(shuō)明
python 3.7
tensorflow 1.5.0
cuda 10.0
cudnn 7.5.0
解決方法
tensorflow框架下設(shè)置GPU按需分配:
config = tf.ConfigProto()
config.gpu_options.allow_growth =True
with tf.Session(config=config)as sess:
keras下設(shè)置GPU按需分配:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
K.set_session(sess)