快捷導(dǎo)航

一招教你解決Pytorch GPU版本安裝慢的問題

更新時(shí)間：2025年03月08日 09:07:15 作者：起個(gè)破名想半天了

Pytorch是一個(gè)流行的深度學(xué)習(xí)框架,廣泛應(yīng)用于計(jì)算機(jī)視覺、自然語言處理等領(lǐng)域,下面我們就來看看如何在Windows操作系統(tǒng)上安裝Pytorch GPU版本吧

Pytorch是一個(gè)流行的深度學(xué)習(xí)框架，廣泛應(yīng)用于計(jì)算機(jī)視覺、自然語言處理等領(lǐng)域。安裝Pytorch GPU版本可以充分利用GPU的并行計(jì)算能力，加速模型的訓(xùn)練和推理過程。接下來，我們將詳細(xì)介紹如何在Windows操作系統(tǒng)上安裝Pytorch GPU版本。

查看是否有顯卡

GPU版本的pytorch需要有顯卡支持,如果沒有顯卡那就只能使用cpu版本了。

cpu版本安裝：

pip install torch torchvision torchaudio

win+s搜索設(shè)備管理器，打開后，點(diǎn)擊顯示適配器

若你的電腦有獨(dú)立顯卡且顯卡版本GTX>10,RTX>20那么便可以使用GPU版本的Pytorch。

顯卡版本小于上述版本的顯卡是沒有CUDA支持的,無法使用Pytorch的GPU版本，只能使用CPU版本。

查看CUDA版本

win+r cmd命令行里輸入：

nvidia-smi

查看CUDA版本：

我這里是12.6,記住這個(gè)版本號(hào)。

下載CUDA驅(qū)動(dòng)

NVIDIA驅(qū)動(dòng)下載:

CUDA Toolkit Archive | NVIDIA Developer

根據(jù)剛剛的CUDA版本選擇,12.6就下載CUDA Toolkit 12.6即可

點(diǎn)擊后,這樣選擇:

然后按部就班安裝即可。

Whl文件下載

裝過pytorch的都知道，用pip或conda命令從官網(wǎng)安裝pytorch可能會(huì)出現(xiàn)runtime error等錯(cuò)誤，安裝成功后可能會(huì)面臨安裝的是cpu版本的pytorch，或者是python版本與pytorch版本不匹配等問題。所以我建議可以從下面鏈接先下載好對(duì)應(yīng)的whl文件，鏈接如下:

https://download.pytorch.org/whl/torch

cu126表示gpu cuda版本，下載好nvidia終端，并添加到環(huán)境變量后輸入nvidia-smi 可以查看gpu版本號(hào) cp312代表python版本號(hào),必須和python版本一致,不然下載后上也無法安裝。wind_amd64表示windows系統(tǒng)

按照自己的版本選擇后點(diǎn)擊即可在瀏覽器里下載

點(diǎn)擊藍(lán)色文件夾，跳轉(zhuǎn)到下載目錄

點(diǎn)擊文件ctrl+shift+c或右鍵復(fù)制文件地址后，win+r輸入cmd打開cmd命令行

輸入pip install后將路徑復(fù)制然后等待下載即可,一般來說1~2分鐘即可下載完畢

驗(yàn)證

import torch
print(f"PyTorch版本: {torch.__version__}")
print(f"CUDA是可用的:{torch.cuda.is_available()}")
print(f"當(dāng)前pytorch使用設(shè)備:{torch.device('cuda' if torch.cuda.is_available() else 'cpu')}")