無法使用GPU運行Tensorflow / Keras

Question

我試圖用我的GPU運行Keras但是出現以下錯誤：

C：\\ Python36 \\ lib \\ site-packages \\ skimage \\ transform_warps.py:84:UserWarning：默認模式'constant'將在skimage 0.15中更改為'reflect'。 警告（“默認模式，'常數'，將在”EC：\\ tf_jenkins \\ workspace \\ rel-win \\ M \\ windows -gpu \\ PY \\ 36 \\ tensorflow \\ stream_executor \\ cuda \\ cuda_dnn.cc中更改為'反映' ：378]

已加載的運行時CuDNN庫：7102（兼容版本7100），但源代碼是使用7003（兼容版本7000）編譯的。 如果使用二進制安裝，請升級您的CuDNN庫以匹配。 如果從源構建，請確保在運行時加載的庫與編譯配置期間指定的兼容版本匹配。

FC：\\ tf_jenkins \\ workspace \\ rel-win \\ M \\ windows -gpu \\ PY \\ 36 \\ tensorflow \\ core \\ kernels \\ conv_ops.cc：717]檢查失敗：stream-> parent（） - > GetConvolveAlgorithms（conv_parameters.ShouldIncludeWinogradNonfusedAlgo（ ），和算法）

我有tensorflow 1.6，CUDA版本：Cuda編譯工具，發布9.0，V9.0.176

有誰知道這里有什么不對嗎？

Answer 1

您需要安裝cuDNN 7.0.5。 該文件可以在這里下載。 單擊“下載”並同意條款后，將列出該選項。