tensorflow - 我应该更改我的 Keras 代码以在 GPU 上运行它吗？

Question

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

标签： tensorflowkerasgpucpu

Answer 1

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

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