Skip to content

Instantly share code, notes, and snippets.

@arjun-kava

arjun-kava/fee_gpu_memory_keras.py

Created August 14, 2018 05:36
Show Gist options
  • Save arjun-kava/f13b8a1b81aa0241d4c3b55b14dd09cb to your computer and use it in GitHub Desktop.
Save arjun-kava/f13b8a1b81aa0241d4c3b55b14dd09cb to your computer and use it in GitHub Desktop.
Download ZIP
Free GPU memory keras
Raw
fee_gpu_memory_keras.py
from keras import backend as K
cfg = K.tf.ConfigProto()
cfg.gpu_options.allow_growth = True
K.set_session(K.tf.Session(config=cfg))
K.clear_session()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment