Skip to content

Instantly share code, notes, and snippets.

@zldrobit

zldrobit/tf-allow-growth.txt

Created March 24, 2020 09:04
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save zldrobit/0b2d0e613bf0faa8a18e0a14aabf1931 to your computer and use it in GitHub Desktop.
Save zldrobit/0b2d0e613bf0faa8a18e0a14aabf1931 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
tf-allow-growth.txt
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.compat.v1.InteractiveSession(config=config)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment