arjun-kava/fee_gpu_memory_keras.py

Created August 14, 2018 05:36

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/arjun-kava/f13b8a1b81aa0241d4c3b55b14dd09cb.js"></script>
Save arjun-kava/f13b8a1b81aa0241d4c3b55b14dd09cb to your computer and use it in GitHub Desktop.

Download ZIP

Revisions

arjun-kava created this gist Aug 14, 2018.

fee_gpu_memory_keras.py

-Original file line number
+Diff line change
@@ -0,0 +1,7 @@
+    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()