1. 程式人生 > 其它 >tensorflow和cuda以及cudnn版本對齊

tensorflow和cuda以及cudnn版本對齊

技術標籤:機器學習入門

起因

今天想用下買了已經很久的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版本一致,親測。