概述
CUDA和cuDNN是由NVIDIA提供的兩個關(guān)鍵軟件庫,，用于利用NVIDIA GPU進(jìn)行高性能計算和深度學(xué)習(xí)加速,。
CUDA是一個由NVIDIA開發(fā)的并行計算平臺和編程模型，用來加速計算密集型任務(wù)
cuDNN是一個GPU加速的深度學(xué)習(xí)庫,，是許多深度學(xué)習(xí)框架（如 TensorFlow,、PyTorch、MXNet 等）中不可或缺的組件,，因為它能夠極大地提高深度學(xué)習(xí)任務(wù)的性能,。
簡單來說，CUDA 是一個通用的GPU編程平臺,，而cuDNN則是基于CUDA之上的一個專門用于深度學(xué)習(xí)的優(yōu)化庫,。
安裝前提：必須有一塊支持CUDA的英偉達(dá)顯卡,，如果電腦沒有英偉達(dá)顯卡，那沒辦法使用哦,。
接下來我將手把手教你們安裝cuda和cudnn
一,、cuda安裝
1.查詢自己電腦可支持最高cuda版本是多少
在終端輸入nvidia-smi命令，查詢可支持的最高cuda版本,，驅(qū)動是向下兼容的,，我電腦可支持最高版本的cuda為12.2，所以安裝cuda版本小于等于12.2的都可以安裝上,。（如果想安裝的cuda版本不在可支持的cuda版本內(nèi),，則可以考慮升級顯卡驅(qū)動，從而實現(xiàn)對cuda高版本的安裝,，去官網(wǎng)下載你電腦對應(yīng)顯卡的驅(qū)動程序：https://www./Download/index.aspx?lang=cn）
顯卡驅(qū)動更新教程如下：注意注意注意：如果你電腦不需要更新顯卡驅(qū)動,，那就跳過顯卡驅(qū)動更新教程哦
1.查看自己電腦顯卡型號
在搜索框輸入設(shè)備管理器并打開
我的型號如下：
2.復(fù)制上面的顯卡驅(qū)動下載地址
查找自己的顯卡型號
3.下載完成直接雙擊安裝
安裝超級簡單，直接下一步下一步就行,，安裝完成之后重啟電腦,，在終端輸入nvidia-smi命令，來查看可支持的最高cuda版本
2.cuda安裝包下載
下載前需要確定自己需要安裝的pytorch版本號是多少,，根據(jù)下面表格來選擇合適版本的cuda和cudnn
cuda與cudnn對應(yīng)的版本表格：
cuda,、CUDAToolkit與pytorch對應(yīng)的版本
點擊這個鏈接進(jìn)行下載：CUDA下載地址
需要注意：30系列顯卡的需要cuda11及以上的版本，我電腦顯卡是3070,，所以需要安裝大于等于cuda11的版本,，我選擇cuda11.3版本的進(jìn)行下載
3.cuda安裝
雙擊安裝包安裝
點擊展開，取消Visual Studio Integration
安裝完成
4.查看cuda是否自動添加到環(huán)境變量里面,，沒有就手動添加
此電腦–>屬性
點擊高級系統(tǒng)設(shè)置–>查看環(huán)境變量
在系統(tǒng)變量,，看到cuda是自動添加到環(huán)境變量里面的，如果你發(fā)現(xiàn)沒有則需要自己手動添加
有的小伙伴說沒有自動生成,，沒有生成的手動添加就行,，
變量如下：
11_3_填自己的cuda版本號，比如你是cuda11.0改成11_0_
CUDA_PATH
CUDA_PATH_V11_3
NVCUDASAMPLES_ROOT
NVCUDASAMPLES11_3_ROOT
值的話修改成自己的路徑,。
5.驗證cuda是否安裝成功
在終端輸入nvcc -V命令,，輸出版本號代表安裝成功
二、cudnn安裝
1.cudnn安裝包下載
下載前需要確定自己需要安裝的pytorch版本號是多少,，根據(jù)下面表格來選擇合適版本的cuda和cudnn
cuda與cudnn對應(yīng)的版本表格：
cuda,、CUDAToolkit與pytorch對應(yīng)的版本
點擊這個鏈接進(jìn)行下載：cudnn下載地址
需要注冊一個賬號才可以下載，隨便注冊就行
需要注意：看表格找到對應(yīng)版本的cudnn,，我安裝cuda版本是11.3,，所以需要安裝8.2.0或者8.2.1都行
找到cudnnv8.2.0，for CUDA 11.x 表示這個cuDNN版本是 CUDA 11.x 版本系列兼容的。CUDA11.x 系列包括了 11.0,、11.1,、11.2 、11.3等具體版本,，x 表示任何小版本號
2.cudnn安裝
cuDNN是以壓縮包的形式提供的
之后找到路徑為：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3的文件夾,，v11.3是我安裝cuda的版本號，你的可能不一樣,，路徑都差不多一樣的
3.驗證cudnn是否安裝成功
cmd進(jìn)入目錄：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\extras\demo_suite
終端輸入：deviceQuery.exe
如果結(jié)果為pass證明安裝成功
到此安裝結(jié)束
總結(jié)
創(chuàng)造不易,，對你有幫助請點個小愛心，謝謝
|
