设置GPU计算
keras使用tensorflow作为后端运行的话,tensorflow是自动识别设备
以下是强制使用CPU或GPU设备。
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(device_count={'gpu':0}))
或者
import tensorflow as tf
import keras.backend.tensorflow_backend as KTF
KTF.set_session(tf.Session(config=tf.ConfigProto(device_count={'gpu':0})))