显存不足

  1. import tensorflow as tf

  2. import numpy as np

  3. import keras

  4. config = tf.compat.v1.ConfigProto(allow_soft_placement=True)

  5.  

  6. config.gpu_options.per_process_gpu_memory_fraction = 0.3

  7. tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))