Skip to content

Instantly share code, notes, and snippets.

@kaecius

kaecius/tensorflow-no-full-memory-allocation.py

Created January 11, 2021 16:02
Show Gist options
  • Save kaecius/a8d5f2330f57da734ff909bb5a791740 to your computer and use it in GitHub Desktop.
Save kaecius/a8d5f2330f57da734ff909bb5a791740 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tensorflow-no-full-memory-allocation.py
import tensorflow as tf # tensorflow==2.4
gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(gpus[0],True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment