【发布时间】:2018-11-09 05:55:56
【问题描述】:
我正在设置我的计算机以使用 GPU 运行 DL,但我找不到有关是否应该安装 keras 或 keras-gpu 的信息。目前我使用 tensorflow-gpu 作为后端运行 conda 和 keras。如果我将 keras 切换到 keras-gpu 会有什么不同?
【问题讨论】:
-
你说的是 conda 包,对吧?你应该澄清你的问题。我相信 keras 没有特定的 gpu 实现,因此您将拥有相同的东西,但在一个包中。
-
是的,我现在在一个小项目中用 keras 和 keras-gpu 测试过,没有发现有什么区别。
-
区别是在cpu或gpu上运行你的代码。
-
@MiloLu 我认为这取决于后端 keras 使用什么,即在我的情况下是 tensorflow 或 tensorflow-gpu。对我来说,如果我使用 keras+tensorflow-gpu 或 keras-gpu+tensorflow-gpu 没有任何区别
标签: tensorflow keras