深度學習GPU加速配置方法
阿新 • • 發佈：2022-12-12
深度學習GPU加速配置方法
一、英偉達官方驅動及工具安裝
- 首先檢查自己的電腦驅動版本，未更新至最新建議先將驅動更新至最新，然後點選Nvidia控制面板
2.在如下介面中點選系統資訊，點選顯示可以看見當前的顯示卡驅動版本，點選元件可以看到紅框中的CUDA版本的最高支援，在安裝時只需要裝這個版本之下的即可。
- 開啟Cuda Tookit的安裝官網，CUDA Toolkit Archive | NVIDIA Developer，選擇自己需要的版本，但不能高於上面的版本號，此處以Cuda Tookit10.2為例，選擇好對應的版本號和系統，進行下載。
-
選擇自定義安裝，注意在安裝這之前至少得有一個Visual Studio，不然會安裝失敗。
-
在自定義中如果是第一次安裝則選擇所有的選項，如果是第n次安裝，則只選擇第一個選項，安裝位置儘量預設。
-
安裝完成後如圖所示：
-
然後在命令列中輸入nvcc -V即可檢視安裝的cuda。
PS:如果未出現考慮環境變數的問題，主要參考以下幾個環境變數，注意安裝位置是預設位置
-
安裝完cuda後，還需安裝cuda的補丁即cuDnn，下載網址cuDNN Download | NVIDIA Developer，因為這裡安裝的是10.2版本的cuda，所以選擇8.7版本的cuDnn for Cuda 10.2，然後根據自己的平臺選擇相應的版本，這裡選擇windows壓縮包。
-
將下載的檔案進行解壓，將這三個檔案進行復制，然後放入安裝目錄（
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2）下的同樣檔案。 -
此時可以開啟pytorch官網，此處以pytorch V1.11.0為例，需要安裝對應cuda版本的pytorch，此處為10.2版本，複製安裝命令即可。
-
注意此處必須是cuda版本，有時候利用換源安裝的是cpu版本，注意區分
-
輸入如下命令，如果最後結果顯示True，說明安裝成功，恭喜可以進行GPU加速了