环境说明
python 3.7
tensorflow 1.5.0
cuda 10.0
cudnn 7.5.0
解决方法
tensorflow框架下设置GPU按需分配:
config = tf.ConfigProto()
config.gpu_options.allow_growth =True
with tf.Session(config=config)as sess:
keras下设置GPU按需分配:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
K.set_session(sess)
网友评论