#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
原因是显存不足,因此设置成按需分配。