????????Pytorch是一個流行的深度學習框架，廣泛應用于計算機視覺、自然語言處理等領域。安裝Pytorch GPU版本可以充分利用GPU的并行計算能力，加速模型的訓練和推理過程。接下來，我們將詳細介紹如何在Windows操作系統上安裝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
https://developer.nvidia.com/cuda-toolkit-archive根據剛剛的CUDA版本選擇,12.6就下載CUDA Toolkit 12.6即可
點擊后,這樣選擇:
?然后按部就班安裝即可。
Whl文件下載
????????裝過pytorch的都知道，用pip或conda命令從官網安裝pytorch可能會出現runtime error等錯誤，安裝成功后可能會面臨安裝的是cpu版本的pytorch，或者是python版本與pytorch版本不匹配等問題。所以我建議可以從下面鏈接先下載好對應的whl文件，鏈接如下:
https://download.pytorch.org/whl/torchhttps://download.pytorch.org/whl/torch
- cu126表示gpu cuda版本，下載好nvidia終端，并添加到環境變量后輸入nvidia-smi ?可以查看gpu版本號?
- cp312代表python版本號,必須和python版本一致,不然下載后上也無法安裝。
- wind_amd64表示windows系統
按照自己的版本選擇后點擊即可在瀏覽器里下載
點擊藍色文件夾，跳轉到下載目錄
點擊文件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')}")
安裝成功！
)
