jupyter-keras-gpu.md

keras(tf backend)でgpuの設定

%matplotlib inline
import tensorflow as tf
from keras import backend as K
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
config.gpu_options.visible_device_list="0"
sess = tf.Session(config=config)
K.set_session(sess)