#GPU按需分配,解决
import tensorflow as tf
import keras
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
keras.backend.tensorflow_backend.set_session(tf.Session(config=config))

参考https://blog.csdn.net/qq_41868689/article/details/98503069

原因是显存不足,因此设置成按需分配。