cuDNN 初始化失敗

Question

tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
  (0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
     [[{{node conv2d_1/convolution}}]]
     [[norm_dense_1/Softmax/_3605]]
  (1) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
     [[{{node conv2d_1/convolution}}]]

嘗試從 Github 運行項目時遇到此錯誤。我已嘗試在新的 conda 環境中從頭開始重新安裝所有程序。 似乎無法弄清楚是什么問題。

GPU Zotac gtx 1070 ti

我目前的軟件安裝如下：-

• Python 3.6
• Tensorflow GPU 1.15
• CUDA 10
• cuDNN 7.4

Answer 1

確保您擁有 CUDA Toolkit 7.5 請打開一個新的 session 然后嘗試加載 tensorflow

你能在沒有任何錯誤的情況下加載 tensorflow 嗎？ 確保正在加載所有 ddl cudard64_101.ddl 和其他文件。 如果 tensorflow 沒有問題，則運行以下命令

physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)

完成后再次嘗試運行代碼

Answer 2

我有同樣的 output。 我的問題是我不小心為 cuda 設置了兩個不同的環境變量。 PATH 有 2 個位置用於 cuda 文件夾，這對我來說是個問題。 刪除其中一個解決了我的問題。