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

Posted

技术标签:

【中文标题】是否有 R 命令使 Keras Tensorflow-GPU 在 CPU 上运行?【英文标题】:Is there R command(s) making Keras Tensorflow-GPU to run on CPU? 【发布时间】:2018-12-10 04:20:22 【问题描述】:

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

【问题讨论】:

【参考方案1】:

让我给你两个答案。

答案#1(正常答案)

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

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

答案#2(好吧,也许你可以“欺骗”keras

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

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

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

【讨论】:

以上是关于是否有 R 命令使 Keras Tensorflow-GPU 在 CPU 上运行?的主要内容,如果未能解决你的问题,请参考以下文章

Keras 预测重复的列

tensorflow -----AttributeError: module ‘tensorflo

为keras输入R转换图像/数组列表

如何在 keras conv 层中进行零填充?

keras 中是不是有基于精度或召回而不是损失的优化器?

忽略 keras 中 R 的缺失目标值的损失函数