Keras 上 GPU 上 windows 与 python 3.7。

Question

最近我不得不完全重新安装我的 python 发行版，由于某些原因，我无法再在 GPU 上运行 keras。

我按照Can I run Keras model on gpu 中的说明进行操作？ 但由于某种原因，我在尝试列出设备时没有看到我的 GPU。

我的版本是：

• tensorflow 和张量流 GPU：2.3.0
• keras：2.3.1
• cudatoolkit：11.3.1

我还没有安装 cudnn，因为说明对我来说有点模糊：我必须将它安装在 cudatoolkit 目录中吗？ 是否需要在 GPU 上运行？

谢谢

Answer 1

好的，这是解决方案：

Python 3.7.13 需要 tensorflow-gpu 2.1.0 来检测 gpu。 所以正确的版本集是：

• Python 3.7.13
• tensorflow 2.1.0
• keras 2.3.0
• cudnn 7.6.5
• cudatoolkit 10.1.243

这样就可以了。 请注意，出于某种原因，在我的情况下，tensorflow 下载了 tensorflow-estimator 2.6.0。 我不得不降级到 2.1.0 才能让 spyder 运行。