如何安装支持 gpu 的 Keras？

Posted 2023-02-24

【中文标题】如何安装支持 gpu 的 Keras？

【英文标题】：How to install Keras with gpu support?

【发布时间】：2019-07-08 09:39:57

【问题描述】：

我为 GPU 安装了 Tensorflow：pip install tensorflow-gpu 但是当我为 Keras pip install keras-gpu 尝试相同的操作时，它给我一个错误：找不到满足要求的版本。

【问题讨论】：

请说明使用的 Python 版本。另请注意，keras-gpu 是 conda 安装，而不是 pip 安装。

【参考方案1】：

要安装具有特定 cuda 版本 9.0 的 tensorflow-gpu，请使用：

“conda install tensorflow-gpu cudatoolkit==9.0 -c anaconda”

keras-gpu 类似

【参考方案2】：

除了以上两个答案，请确保您的 TensorFlow/Keras 环境使用 Python 3.6。截至 2019 年 5 月 10 日，Keras/TensorFlow 不能很好地与 Python 3.7 配合使用。

我尝试在 Python 3.7 中使用 Keras/TensorFlow，但最终不得不重新安装 Anaconda，因为它破坏了我的 Anaconda Prompt。

【参考方案3】：

没有任何keras-gpu 包[更新：现在有，请参阅下面的其他答案]； Keras 是一些后端的封装，包括 Tensorflow，这些后端可能有不同的版本，比如tensorflow 和tensorflow-gpu。但这不适用于 Keras 本身，它应该简单地安装

pip install keras

独立于使用的任何后端（请参阅PyPi docs）。

此外，由于您也将问题标记为anaconda，请注意通常不建议将您的包管理器（即pip 与conda）混合使用，您最好安装Keras 来自Anaconda cloud 与

conda install -c conda-forge keras 

最后，您可能还想知道最新版本的 Tensorflow 包含 Keras 作为子包，因此您无需任何额外安装即可使用它；见https://www.tensorflow.org/guide/keras

要从Anaconda cloud 安装tensorflow-gpu，您应该使用

conda install -c anaconda tensorflow-gpu 

在安装 Keras 之前。确保您在不同的虚拟环境中执行此操作，或者在卸载其他版本（即pip-安装的版本）之后执行此操作，否则有reported problems。

【讨论】：

所以改为 pip install tensorflow-gpu .. 我可以在 anaconda 中安装什么

@SaiKrishnadas 查看更新的答案（无论如何，这是一个简单的谷歌搜索......）；如果答案有帮助，请接受 - 谢谢

Still.. 系统信息 * Windows 7 * pip install tensorflow-gpu * Python版本： 3.6.6 * 使用 virtualenv 安装？点子？ conda?：pip * CUDA/cuDNN 版本：CUDA 8.0 cuDNN 7.1 * GPU 型号和内存：型号：Quadro 4000，内存：18GB 已安装使用 pip install tensorflow-gpu （没有错误，完美安装）在导入 tensorflow 时，出现错误：Failed to load native tensorflow .ImportError: DLL load failed: The specified module could not be found跨度>

您最初的问题是关于keras-gpu，已在答案中解决。如果您对不同的软件包有不同的问题，非常欢迎您在接受这个问题后提出一个新问题。

【参考方案4】：

添加到上面的答案中，这是推荐使用 Anaconda 包管理器的正确答案，但已经过时了 现在有一个 keras-gpu在Anaconda Cloud上打包。

所以一旦你安装了 Anaconda，你只需要创建一个要安装 keras-gpu 的新环境并执行命令：

conda install -c anaconda keras-gpu

这将安装 Keras 以及 tensorflow 和 tensorflow-gpu 库作为后端。 （也无需单独安装 CUDA 运行时和 cudnn 库，因为它们也包含在包中 - 在 Windows 10 上测试并正常工作）。

【讨论】：

在这种情况下，keras 使用tensorflow CPU 版本作为后端，根本不使用 GPU。知道如何解决这个问题吗？

先安装 tensorflow-gpu 然后安装 keras-gpu 成功了！ ***.com/a/69498829/3337089