Keras和Keras-GPU有什么区别?

5

我正在设置我的电脑以使用GPU运行DL,但我找不到有关安装keras还是keras-gpu的信息。目前,我已经使用conda和keras以tensorflow-gpu为后端运行。如果我将keras切换到keras-gpu,会有什么区别呢?


1
你是在说conda包,对吗?你应该澄清你的问题。我相信keras没有特定的gpu实现,所以你会得到同样的东西,但在一个单一的包中。 - Daniel GL
是的,我现在已经在一个小项目中使用keras和keras-gpu进行了测试,并没有发现任何区别。 - Johan
区别在于在 CPU 或 GPU 上运行您的代码。 - Milo Lu
1
@MiloLu 我认为这取决于Keras使用的后端,例如在我的情况下是TensorFlow或TensorFlow-GPU。对我来说,使用keras+tensorflow-gpu或keras-gpu+tensorflow-gpu没有任何区别。 - Johan
1个回答

2
这是从维基百科借来的一段话:
Keras被设计为一个接口而不是独立的机器学习框架。它提供了更高级别、更直观的抽象,使得开发深度学习模型变得容易,无论使用哪种计算后端。
https://en.wikipedia.org/wiki/Keras
只要Keras的三个后端(TensorFlow、Microsoft Cognitive Toolkit或Theano)都不支持GPU版本,使用keras-gpu可能会给您带来麻烦。

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接