cuDNN是由NVIDIA開發(fā)的一個(gè)深度學(xué)習(xí)GPU加速庫,用以輔助NVIDIA的并行計(jì)算架構(gòu)CUDA進(jìn)行一系列的復(fù)雜變換、優(yōu)化等數(shù)學(xué)計(jì)算。因?yàn)橹皇且粋€(gè)輔助庫,安裝cuDNN之前必須先安裝CUDA。CUDA是一個(gè)偉大的架構(gòu),起初只是用于顯卡的圖形處理計(jì)算,因?yàn)槠涑咝阅艿牟⑿行?,逐漸被用來處理各種非常耗時(shí)的可并行化的數(shù)學(xué)計(jì)算,取得了巨大的成功。
安裝visual studio
1、在安裝之前一定要先安裝VS,否則在安裝CUDA過程中就會出現(xiàn)如下情況,從而無法正常安裝
2、登錄自己的賬號,最好下載2017版本的,實(shí)在找不到選擇2019或者2022版本的下載均可,注意一定選擇Community免費(fèi)社區(qū)版。
3、下載完后,雙擊安裝
4、在安裝選項(xiàng)中勾選“C++桌面開發(fā)”即可
5、點(diǎn)擊啟動(dòng),登錄自己的賬號初始化完畢即可
6、啟動(dòng)完畢,關(guān)閉VS即可
安裝CUDA
查看CUDA版本
1、在搜索框輸入NVIDIA,點(diǎn)擊NVIDIA控制面板
2、選中“管理3D設(shè)置”,點(diǎn)擊下方的“系統(tǒng)信息”
3、點(diǎn)擊“組件”按鈕,查看CUDA版本
4、可以看到我這里是CUDA 12.1
下載并安裝CUDA
1、進(jìn)入CUDA官網(wǎng),根據(jù)上面確定的CUDA版本找到對應(yīng)版本
2、選擇自己的版本,Installer Type 方式選擇exe(local) , 之后點(diǎn)擊“Download”按鈕
3、雙擊下載的exe安裝包
4、選擇臨時(shí)解壓位置,點(diǎn)擊“OK”,等待解壓完畢
5、等待系統(tǒng)檢測完畢,點(diǎn)擊同意并繼續(xù)
6、選擇自定義安裝,點(diǎn)擊下一步
7、繼續(xù)點(diǎn)擊“下一步”
8、選擇安裝位置,一定要記住,之后需要用到,一般默認(rèn)即可
9、等待安裝完成
10、點(diǎn)擊“下一步”
11、安裝完畢,取消勾選,點(diǎn)擊關(guān)閉即可
12、查看系統(tǒng)環(huán)境變量,點(diǎn)擊設(shè)置—>關(guān)于—>高級系統(tǒng)設(shè)置
13、查看系統(tǒng)環(huán)境變量
14、可以看到,CUDA環(huán)境變量已經(jīng)自動(dòng)加入到系統(tǒng)中
測試CUDA是否安裝成功
1、win+R 快捷鍵,之后輸入cmd,調(diào)出命令行終端
2、在命令行輸入nvcc -V【返回版本號則安裝成功】
安裝cuDNN
1、進(jìn)入CUDNN官網(wǎng),需要注冊/登錄后才可以下載
2、登錄后,選擇我們與cuda對應(yīng)的版本下載安裝即可
3、將下載的壓縮包解壓
4、將解壓后bin目錄的內(nèi)容全部放到CUDA對應(yīng)的bin目錄下
5、將解壓后include目錄下的內(nèi)容全部放到CUDA對應(yīng)的include目錄下
6、將解壓后lib目錄下x64目錄中的內(nèi)容全部放到CUDA對應(yīng)的lib目錄下的x64目錄中
7、打開系統(tǒng)環(huán)境變量,步驟同上
8、編輯Path變量
9、將這四條添加進(jìn)去,前兩條在之前已經(jīng)自動(dòng)添加了,點(diǎn)擊“新建”添加后兩條即可
測試CUDNN是否安裝成功
通過cmd打開命令行終端,步驟同上,在終端輸入nvidia-smi,返回GPU型號則安裝成功
上一篇:Scratch電腦版