Unable To Create Session In Jupyter Notebook For Keras Kfold Validation Code
I wrote a code in jupyter notebook using keras/tensorflow. The code worked well when I ran it on my laptop that runs tensorflow on cpu but when I run the same code on my home compu
Solution 1:
Placed at beginning of my code:
from keras import backend as K
cfg = K.tf.ConfigProto()
cfg.gpu_options.allow_growth = True
K.set_session(K.tf.Session(config=cfg))
credit: http://forums.fast.ai/t/tip-clear-tensorflow-gpu-memory/1979/7
Post a Comment for "Unable To Create Session In Jupyter Notebook For Keras Kfold Validation Code"