cuDNN是由NVIDIA開發(fā)的一個深度學習GPU加速庫,用以輔助NVIDIA的并行計算架構CUDA進行一系列的復雜變換、優(yōu)化等數(shù)學計算。因為只是一個輔助庫,安裝cuDNN之前必須先安裝CUDA。CUDA是一個偉大的架構,起初只是用于顯卡的圖形處理計算,因為其超高性能的并行性,逐漸被用來處理各種非常耗時的可并行化的數(shù)學計算,取得了巨大的成功。
安裝visual studio

1、在安裝之前一定要先安裝VS,否則在安裝CUDA過程中就會出現(xiàn)如下情況,從而無法正常安裝
2、登錄自己的賬號,最好下載2017版本的,實在找不到選擇2019或者2022版本的下載均可,注意一定選擇Community免費社區(qū)版。
3、下載完后,雙擊安裝
4、在安裝選項中勾選“C++桌面開發(fā)”即可
5、點擊啟動,登錄自己的賬號初始化完畢即可
6、啟動完畢,關閉VS即可
安裝CUDA

查看CUDA版本
1、在搜索框輸入NVIDIA,點擊NVIDIA控制面板
2、選中“管理3D設置”,點擊下方的“系統(tǒng)信息”
3、點擊“組件”按鈕,查看CUDA版本
4、可以看到我這里是CUDA 12.1
下載并安裝CUDA
1、進入CUDA官網(wǎng),根據(jù)上面確定的CUDA版本找到對應版本
2、選擇自己的版本,Installer Type 方式選擇exe(local) , 之后點擊“Download”按鈕
3、雙擊下載的exe安裝包
4、選擇臨時解壓位置,點擊“OK”,等待解壓完畢
5、等待系統(tǒng)檢測完畢,點擊同意并繼續(xù)
6、選擇自定義安裝,點擊下一步
7、繼續(xù)點擊“下一步”
8、選擇安裝位置,一定要記住,之后需要用到,一般默認即可
9、等待安裝完成
10、點擊“下一步”
11、安裝完畢,取消勾選,點擊關閉即可
12、查看系統(tǒng)環(huán)境變量,點擊設置—>關于—>高級系統(tǒng)設置
13、查看系統(tǒng)環(huán)境變量
14、可以看到,CUDA環(huán)境變量已經(jīng)自動加入到系統(tǒng)中
測試CUDA是否安裝成功
1、win+R 快捷鍵,之后輸入cmd,調(diào)出命令行終端
2、在命令行輸入nvcc -V【返回版本號則安裝成功】
安裝cuDNN

1、進入CUDNN官網(wǎng),需要注冊/登錄后才可以下載
2、登錄后,選擇我們與cuda對應的版本下載安裝即可
3、將下載的壓縮包解壓
4、將解壓后bin目錄的內(nèi)容全部放到CUDA對應的bin目錄下
5、將解壓后include目錄下的內(nèi)容全部放到CUDA對應的include目錄下
6、將解壓后lib目錄下x64目錄中的內(nèi)容全部放到CUDA對應的lib目錄下的x64目錄中
7、打開系統(tǒng)環(huán)境變量,步驟同上
8、編輯Path變量
9、將這四條添加進去,前兩條在之前已經(jīng)自動添加了,點擊“新建”添加后兩條即可
測試CUDNN是否安裝成功
通過cmd打開命令行終端,步驟同上,在終端輸入nvidia-smi,返回GPU型號則安裝成功
上一篇:Scratch電腦版