Skip to content

Instantly share code, notes, and snippets.

@fo40225

fo40225/keras_dynamic_vram.py

Created August 11, 2018 09:15
Show Gist options
  • Save fo40225/e1e06b0f0cc9de991ac99fedb9e7dee7 to your computer and use it in GitHub Desktop.
Save fo40225/e1e06b0f0cc9de991ac99fedb9e7dee7 to your computer and use it in GitHub Desktop.
Download ZIP
keras dynamic vram
Raw
keras_dynamic_vram.py
from keras import backend as K
import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)
K.set_session(session)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment