tensorflow - 如何在 gpu 支持下安装 Keras？

Question

我使用以下命令为 GPU 安装了 Tensorflow: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 上测试并正常工作)。