關於 tensorflow-gpu 中 CUDA 和 CuDNN 版本適配問題

問題

　　今天在使用 tensorflow-yolov3 的時候，發現報錯

Loaded runtime CuDNN library: 7.1.4 but source was compiled with: 7.2.1. CuDNN library major and minor version needs to match or have higher minor version in case of 
CuDNN 7.0 or later version.

emmmmmm。。。看了下，好像說是CuDNN的版本有點低，說是源文件是用 CuDNN 7.2.1 版本編譯的，而咱們使用的是版本 7.1.4，這可能跟咱們安裝的 tensorflow-gpu 的版本有關

解決方法

我到 stackoverflow 上查找了相關錯誤，發現大體有兩種解決思路

1.下降tensorflow-gpu的版本

這個方法我是在這篇博客上看到的，我沒去嘗試，我本身使用的是方法二。

2.替換CuDNN的版本

去英偉達官網上下載7.2.1版本之後的與本機安裝的 CUDA 版本適配的 CUDNN ，我下的是如今最新版的 7.4.1 的版本

替換好以後再次運行以前的代碼，發現運行成功。

3.從源代碼本身從新構建

感受有點高端。。沒去試，仍是本身太菜了，能簡單解決就簡單解決吧 orz。。。

CUDA版本問題

感受這個也須要題一題，上次我把臺式機裝成雙系統以後，在 linux 環境下配置 CUDA ，跟我以前一篇博客的流程差很少，結果運行就是錯誤，後來查了才發現我裝的是 CUDA10.0，而 tensorflow-gpu 並不支持 CUDA10.0，只支持 8.0 和 9.0 的版本。。。。簡直了

這裏附上 tensorflow 官網上給出的 tensorflow-gpu 對應的 CUDA 版本

Linux

macOS

Windows

 

　　但願以上內容能幫助到你