keras gpu config session

import keras

config = tf.ConfigProto( device_count = {'GPU': 0 , 'CPU': 30} )

sess = tf.Session(config=config) 

keras.backend.set_session(sess)

model.fit(x_train, y_train, epochs=epochs, validation_data=(x_test, y_test))