一、安裝CUDA
1.1、下載安裝包
cuda可以認為就是Nvidia為了顯卡煉丹搞的一個軟件,其下載地址為:CUDA Toolkit 12.2 Update 1 Downloads | NVIDIA Developer
當(dāng)你點進這個鏈接的時候,你需要依次選擇
1是選擇系統(tǒng),這里選windows
2是選擇平臺,32位還是64位,其實只支持64位
3是選擇系統(tǒng)版本,這里我們選擇win10,如果是其他的系統(tǒng)就選擇其他的
4是選擇本地下載,也就是一次全部下載完然后安裝,還是網(wǎng)絡(luò)下載,即下載一個安裝工具,邊下載邊安裝
5是下載,可以看到安裝包有3GB
需要注意: 這里我們直接選擇了最新版的CUDA,如果你需要別的版本,直接點擊這個鏈接 CUDA Toolkit Archive | NVIDIA Developer
這里面有你需要的版本。
1.2、安裝
雙擊下載好的文件
后面就一路確定就好了,默認的安裝位置在
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2
所以我們需要添加兩個系統(tǒng)變量,實際上在安裝的時候已經(jīng)默認添加了,為了防止沒有添加成功,這里我們可以看一下
點擊屬性


主要看有沒有這兩條,沒有的話就新建添加一下。
二、CUDNN安裝
2.1、CUDNN下載
下載地址:cuDNN Download | NVIDIA Developer
你要同意他的政策,也就是點擊前面的小方塊
然后就會出現(xiàn)兩個下載選項,我們選擇第一個,因為第一個CUDNN對應(yīng)的CUDA版本是12,我們安裝的CUDA版本是12。
如果你需要下載歷史版本,點擊這個鏈接 cuDNN Archive | NVIDIA Developer
需要注意: 也許你進這個網(wǎng)站需要登錄,這是正常的,我因為之前登錄過了,所以沒有遇到,需要登錄的話就注冊個賬號然后登錄就好了。
2.2、安裝
下載好了是一個壓縮包,我們解壓到當(dāng)前為止可以得到三個文件夾

把這三個文件夾復(fù)制到我們之前的那個CUDA目錄下,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2
,這個目錄下也有三個同名文件,把文件復(fù)制過來就可以了。
復(fù)制完成后,需要添加環(huán)境變量,這次我們添加到PATH中,

2.3、驗證
怎么知道我們安裝的CUDA和CUDNN是正確的呢,需要驗證一下,打開CMD,輸入命令
nvcc -V

有這些版本號之類的東西基本上驗證安裝成功了,打開這個地址可以進一步驗證
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\extras\demo_suite
運行這兩個程序
得到下面兩個結(jié)果

驗證安裝成功
三、Pytorch安裝
3.1、安裝Anaconda
可以參考我這篇博客Anaconda萌新入門_LyaJpunov的博客-CSDN博客
就是下載一個程序,然后一路確認就可以了。現(xiàn)在建議從官方下載,官方下載也很快了已經(jīng)Free Download | Anaconda
3.2、創(chuàng)建并且激活虛擬環(huán)境
進入命令行后,創(chuàng)建一個名為learning的虛擬環(huán)境
conda create -n learning
然后激活這個環(huán)境
conda activate learning
3.3、安裝pytorch
我們進入Pytorch的下載地址Start Locally | PyTorch,然后發(fā)現(xiàn),我們裝的是最新的CUDA12.2版本,但是pytorch最新的穩(wěn)定版本是支持到11.7,太干了,但是經(jīng)過查找資料發(fā)現(xiàn)是可以用的。
當(dāng)然我們也可以使用預(yù)覽版本,也就是不穩(wěn)定版本,它支持到了12.1
我們還是按照CUDA11.7來吧,輸入以下命令,這個命令也是網(wǎng)站生成的
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
經(jīng)過一段時間就安裝好了,安裝好了以后可以通過Python腳本驗證一下文章來源:http://www.zghlxwxcb.cn/news/detail-637932.html
import torch
torch.cuda.is_available()
如果輸出True就沒有問題文章來源地址http://www.zghlxwxcb.cn/news/detail-637932.html
到了這里,關(guān)于安裝CUDA與CUDNN與Pytorch(最新超級詳細圖文版本2023年8月最新)的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!