Skip to content

Instantly share code, notes, and snippets.

@t-ae

t-ae/restrict_memory.py

Created March 29, 2017 04:50
Show Gist options
  • Save t-ae/a7fd0123b73e0b2368c83390e1161c7d to your computer and use it in GitHub Desktop.
Save t-ae/a7fd0123b73e0b2368c83390e1161c7d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
restrict_memory.py
import tensorflow as tf
from keras import backend as K
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)
K.set_session(sess)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment