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 運行。