tensorflow和cuda以及cudnn版本對齊
阿新 • • 發佈:2021-01-12
技術標籤:機器學習入門
起因
今天想用下買了已經很久的GPU來玩tensorflow,畢竟顯示卡也不能只是用於玩遊戲吧。然後發現因為cuda 和cudnn版本問題要解除安裝掉舊的重灌新的,踩了一天的坑,來記錄一下。
cuda版本問題
在NVIFIA CONTROL PANEL裡顯示,我應該裝cuda 11.2版本,但是官網上最新版本是11.1。這就有點坑人了。cuda 11.2的download網址如下:cuda 11.2
登入NVIDIA網站問題
登入NVIDIA網站的時候,驗證碼的問題老是被攔截,但是無法登入,解決方法有兩個:
1.不要開任何代理軟體
2.用UU加速器可以(是的你沒聽錯UU加速器)
NVSMI資料夾丟失問題
在C:\Program Files\NVIDIA Corporation資料夾下沒有NVSMI資料夾,這個時候不要去下載NVSMI檔案,如果下載了NVSMI檔案放在C:\Program Files\NVIDIA Corporation中,執行nvidia-smi會顯示
Failed to initialize NVML: Unknown Error
其實nvidia-smi在
C:\Windows\System32\DriverStore\FileRepository\nvdm*\nvidia-smi.exe
這個目錄下,進入對應目錄執行nivdia-smi就可以了
cudnn版本問題
CUDA11.2 對應的cudnn版本與CUDA11.1對應的cudnn版本一致,親測。