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

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

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

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

查看是否有顯卡

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

cpu版本安裝：

pip install torch torchvision torchaudio

win+s搜索設備管理器，打開后，點擊顯示適配器

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

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

查看CUDA版本

win+r cmd命令行里輸入：

nvidia-smi

查看CUDA版本：

我這里是12.6,記住這個版本號。

下載CUDA驅動

NVIDIA驅動下載:

CUDA Toolkit Archive | NVIDIA Developer

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

點擊后,這樣選擇:

然后按部就班安裝即可。

Whl文件下載

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

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

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

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

點擊藍色文件夾，跳轉到下載目錄

點擊文件ctrl+shift+c或右鍵復制文件地址后，win+r輸入cmd打開cmd命令行

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

驗證

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