gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
tf.config.experimental.set_memory_growth(gpu, True)
Preventing Keras to allocate unnecessary GPU memory
Talles L - Aug 19 - Talles L - Aug 19 - Talles L - Aug 19 - gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
tf.config.experimental.set_memory_growth(gpu, True)
