如何安装支持GPU的Keras？

Question

我使用 pip install tensorflow-gpu 安装了GPU版本的Tensorflow，但是当我尝试使用同样的方法安装Keras的GPU版本pip install keras-gpu时，出现了错误：找不到能满足要求的版本。

- Sai Krishnadas

---

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

Answer 1

在下面的答案基础上补充一点，推荐使用Anaconda软件包管理器是正确的，但已经过时了，因为现在Anaconda Cloud上有一个keras-gpu软件包。

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

conda install -c anaconda keras-gpu

这将安装Keras以及作为后端的tensorflow和tensorflow-gpu库。(不需要单独安装CUDA运行时和cudnn库，因为它们也包含在软件包中 - 在Windows 10上测试过并且可以工作)。

- Gus Long

---

在这种情况下，keras 使用的是 tensorflow 的 CPU 版本作为后端，并且根本没有使用 GPU。有什么解决方法吗？ - Nagabhushan S N

先安装tensorflow-gpu，然后再安装keras-gpu就可以了！https://dev59.com/Z8Lra4cB1Zd3GeqPH1wP#69498829 - Nagabhushan S N

Answer 2

目前还没有 keras-gpu 包。Keras 是一个包装器，可以使用多种后端，其中包括 Tensorflow，这些后端可能有不同的版本，例如 tensorflow 和 tensorflow-gpu。但是对于 Keras 本身来说并不适用，应该使用以下命令简单安装：

pip install keras

无论使用什么后端（请参见PyPi文档），都可以独立运行。

此外，由于您已将问题标记为anaconda，因此请注意通常不建议混合使用软件包管理器（例如pip和conda），最好从Anaconda云安装Keras。

conda install -c conda-forge keras 

最后，您可能也会感兴趣了解最近版本的Tensorflow将Keras作为子包含在内，因此您可以在不进行任何其他安装的情况下使用它；请参见https://www.tensorflow.org/guide/keras

如果要从Anaconda云安装tensorflow-gpu，应该使用以下命令：

conda install -c anaconda tensorflow-gpu 

安装 Keras 之前，请确保在不同的虚拟环境中进行安装，或者在卸载其他版本（例如使用pip安装的版本）后再安装，因为否则可能会出现已报告的问题。

- desertnaut

---

那么，如果不使用pip install tensorflow-gpu，我可以用什么来在Anaconda中安装它？ - Sai Krishnadas

@SaiKrishnadas请查看更新的答案（不管怎样，这是一个简单的谷歌搜索...）; 如果答案有帮助，请友善地接受它-谢谢 - desertnaut

仍然.. 系统信息

• Windows 7
• pip install tensorflow-gpu
• Python 版本：3.6.6
• 安装方式：使用 virtualenv? pip? 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 - Sai Krishnadas

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

Answer 3

除了上面两个答案之外，确保您的TensorFlow/Keras环境使用Python 3.6。截至2019年5月10日，Keras/TensorFlow与Python 3.7的兼容性不是很好。

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

- Sean McCarthy

Answer 4

要安装特定的CUDA版本9.0的tensorflow-gpu，请使用以下命令：

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

同样适用于keras-gpu。

- SKG

---

这可能会与其他更新的CUDA软件包（如PyCUDA）发生冲突。 - lefunction

如果那些包在不同的conda环境中，你不应该遇到任何问题。 - alercelik