如何在Keras中利用CPU和GPU的所有核心

时间:2017-12-22 13:39:42

标签: tensorflow deep-learning keras keras-layer

我是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的所有内核?

0 个答案:

没有答案