【发布时间】：2019-12-02 23:24:02
【问题描述】：
假设我在基于 CPU 的 tensorflow 上编写了我的 keras 模型、训练和测试，现在我决定使用 GPU，除了上面提到的 4 点之外，我是否应该改变我的代码编写方式？能够在GPU上编译？？？或者我的 keras 代码可以保持不变并且仍然可以在 GPU 上运行。谢谢
【问题讨论】：
-
一切正常。
标签： tensorflow keras gpu cpu
【发布时间】：2019-12-02 23:24:02
【问题描述】：
它应该只在 GPU 上运行，通常只有环境要求才能在 GPU 上运行，例如启用 gpu 的后端 (tensorflow-gpu)、安装 CUDA/CuDNN 等。
这当然假设您的代码不会强制模型使用 CPU，例如，使用 with tf.device("cpu")，或使用没有 GPU 实现的层。
【讨论】：