gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
tf.config.experimental.set_memory_growth(gpu, True)
Talles L
Posted on
Preventing Keras to allocate unnecessary GPU memory
For further actions, you may consider blocking this person and/or reporting abuse
Top comments (0)