如何在gpu支持下安装Keras？

Question

我安装Tensorflow为GPU使用：pip install tensorflow-gpu 但是，当我尝试了同样的Keras pip install keras-gpu，它把我拉到一个错误：找不到版本满足要求。

Answer 1

在建议使用Anaconda软件包管理器方面，上述答案是正确的答案，但过时的是，Anaconda Cloud上现在有了keras-gpu软件包。

因此，一旦安装了Anaconda，您只需要创建一个新环境就可以在其中安装keras-gpu并执行以下命令：

conda install -c anaconda keras-gpu

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

Answer 2

有没有任何keras-gpu包装[ 更新：现在有，见下面的其他答案] Keras是围绕某些后端（包括Tensorflow）的包装，这些后端可能有不同的版本，例如tensorflow和tensorflow-gpu。但这不适用于Keras本身，应与

pip install keras

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

此外，由于您也将问题标记为anaconda，因此通常不建议混用软件包管理器（例如pip与conda），并且最好从Anaconda云中安装Keras 并使用

conda install -c conda-forge keras 

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

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

conda install -c anaconda tensorflow-gpu 

在安装Keras之前。确保已在其他虚拟环境中执行此操作，或者在卸载了其他版本（pip即已安装的版本）之后进行了此操作，否则可能会出现其他问题。