是否有 R 命令使 Keras Tensorflow-GPU 在 CPU 上运行？

Question

我在 R 中运行 Keras 并使用 Tensorflow-GPU 后端。是否可以在不重新安装后端的情况下强制 Keras 在 CPU 上运行？

Answer 1

让我给你两个答案。

答案#1（正常答案）

不，很遗憾没有。对于keras，CPU 和 GPU 是 2 个不同的版本，您可以在安装时从中选择。

您似乎记得您在安装时选择了 GPU。我猜你希望你只是设置一个次要选项，而不是选择程序的一个版本。很遗憾，您选择了要安装的 keras 版本。

答案#2（好吧，也许你可以“欺骗”keras）

您似乎可以使用环境变量值来欺骗keras，让您认为您的 CPU 就是您的 GPU。

这看起来可能会产生意想不到的结果，但it seemed to work 这些 Python 用户。

我不会担心他们使用 Python 的事实。他们只是使用他们的语言来设置环境变量。所以你可以在 R 中做同样的事情 或直接在您的操作系统中。