如何强制keras使用tensorflow GPU后端

Question

我知道这是最受欢迎的问题之一，但到目前为止，没有一个解决方案对我有用。

tensorflow v1.13.1我正在运行用和编写的遗留代码keras v2.2.4。我无法修改代码来运行最新的张量流版本。由于 keras 现已合并到tensorflow中，因此我在通过pip安装特定版本的tensorflow和keras时遇到问题。我发现anaconda可以选择使用上述版本安装keras和tensorflow。所以，我安装了它

conda install -c conda-forge keras-gpu=2.2.4 tensorflow-gpu=1.13.1

它安装了该版本并且一切正常。但它不使用 GPU，而是在 CPU 上运行。我注意到 anaconda 安装了 Tensorflow 的 CPU 和 GPU 版本，我想这就是它默认为 CPU 版本的原因。所以，我的问题是，如何强制它使用 GPU 版本？

PS：有很多答案建议删除CPU版本的tensorflow。但是当我尝试删除 CPU 版本时，conda 会卸载所有内容，包括 keras。所以，我认为当两者都安装时应该有一种使用tensorflow-gpu的方法。任何在这方面的帮助表示赞赏！

Answer 1

先安装tensorflow，然后安装keras就可以了！

conda install tensorflow-gpu=1.13.1
conda install keras-gpu=2.2.4