我可以在GPU上运行Keras模型吗？

是的，您可以在GPU上运行keras模型。几件事您将必须首先检查。

1. 您的系统具有GPU（Nvidia。因为AMD尚未运行）
2. 您已经安装了Tensorflow的GPU版本
3. 您已安装CUDA安装说明
4. 验证Tensorflow是否与GPU一起运行，检查GPU是否正常工作

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

要么

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

输出将是这样的：

[  name: "/cpu:0"device_type: "CPU",  name: "/gpu:0"device_type: "GPU"]

完成所有这些操作后，您的模型将在GPU上运行：

要检查keras（> = 2.1.1）是否使用GPU：

from keras import backend as KK.tensorflow_backend._get_available_gpus()

祝一切顺利。