Skip to content Skip to sidebar Skip to footer

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"