深度學習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加速了
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/539777.html
標籤：其他