Tensorflow GPU 正在工作,但不再工作錯誤 - 無法取得卷積演算法。這可能是因為 cuDNN 初始化失敗

Tensorflow GPU 正在工作,但不再工作錯誤 - 無法取得卷積演算法。這可能是因為 cuDNN 初始化失敗

所以我安裝了 cuda 10 和tensorflow gpu 1.13.1,一切都工作正常,直到我導入了 VGG16 模型。運行模型一次並且一切正常後,當嘗試在 keras 中運行另一個模型時,我收到此錯誤。

取得卷積演算法失敗。這可能是因為 cuDNN 初始化失敗,因此請嘗試查看上面是否列印了警告日誌訊息。 [[{{節點 conv2d_1/卷積}}]] [[{{節點損失/mul}}]]

現在,每次我嘗試在 keras 中運行任何模型時,都會遇到相同的錯誤。

相關內容