python - 如何在 GPU 上使用 Keras？

我已经成功安装了带有 GPU 的 TensorFlow。当我运行以下脚本时，我得到了这个结果:

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

例如，如果我在 Keras 中运行 CNN，它会自动使用 GPU 吗？或者我是否必须编写一些代码来强制 Keras 使用 GPU？

例如，对于 MNIST 数据集，我将如何使用 GPU？

model = Sequential()
model.add(Conv2D(32, kernel_size=(3, 3),
                 activation='relu',
                 input_shape=input_shape))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(num_classes, activation='softmax'))

model.compile(loss=keras.losses.categorical_crossentropy,
              optimizer=keras.optimizers.Adadelta(),
              metrics=['accuracy'])

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

最佳答案

您不必明确告诉 Keras 使用 GPU。如果 GPU 可用(并且从您的输出中我可以看到是这种情况)，它将使用它。

您还可以通过查看模型训练期间 GPU 的使用情况来凭经验进行检查:如果您使用的是 Windows 10，则只需打开任务管理器并查看“性能”选项卡(请参阅 here )。