我是否应该更改Keras代码以在GPU上运行它？

Question

假设我在基于CPU的Tensorflow上编写了我的keras模型，训练和测试，现在我决定使用GPU（除了上述4点之外），我是否应该更改代码编写方式？为了能够在GPU上进行编译？？？或者，我的keras代码可以保持不变，并且仍可以在GPU上运行。谢谢

Answer 1

它应该只能在GPU上运行，通常只有在环境要求下才能在GPU上运行，例如安装启用gpu的后端（tensorflow-gpu），安装CUDA / CuDNN等。

这当然假设您的代码不会将模型强制使用CPU，例如，使用with tf.device("cpu")或使用没有GPU实现的图层。