如何在CPU和GPU上同时运行Keras？

Question

我正在训练一个模型，该模型由CNN和RNN组成，并具有不同的输入，其输出将被馈送到FCN。 现在的问题是：

是否可以在CPU上并行运行CNN和在GPU上并行运行RNN？

如Tensorflow中所示：

with tf.device('/cpu:0'):
    CNN...
with tf.device('/gpu:0'):
    RNN...

Answer 1

您可以完全使用您发布的代码。

选择设备，定义层，选择另一个设备，定义其他层：

现在，最佳性能可能会使用相反的方法：

with tf.device('/gpu:0'):
    CNNKerasLayers....
with tf.device('/cpu:0'):
    RNNKerasLayers....

现在，我不确定keras将如何处理（有效地并行处理或不并行处理），但很可能只是让tensorflow解决。