我是Keras和TensorFlow的新人。我在keras中准备了一个非常小的卷积神经网络,TensorFlow作为后端。
model = Sequential()
inputShape = (height, width, depth)
if K.image_data_format() == "channels_first":
inputShape = (depth, height, width)
model.add(Conv2D(32, (3, 3), padding="same",
input_shape=inputShape))
model.add(Activation("relu"))
# softmax classifier
model.add(Flatten())
model.add(Dense(classes))
model.add(Activation("softmax"))
我的机器有外置GPU。当我训练模型时,它使用GPU而不是CPU核心。如何使其能够使用CPU和GPU的所有内核?