Win如何驗(yàn)證Pytorch正確了CUDA

1、啟動(dòng)Python控制臺(tái)： 如果你使用的是新版本的Anaconda，只需在Anaconda的控制窗口中輸入python即可打開(kāi)Python控制臺(tái)。 檢查PyTorch情況： 在Python控制臺(tái)中輸入命令import torch。 如果沒(méi)有出現(xiàn)任何錯(cuò)誤，說(shuō)明PyTorch已經(jīng)成功。

2、方法一：，直接在Pytorch版本；Run this Command （2）方法二：離線，在此相應(yīng)包；找到自己需要的torch版本進(jìn)行，例如：torch-10+cu117-cp37-cp37m-win_amd6whl 代表的是torch 10 版本，CUDA 17 版本，Python 7 版本。

3、首先需要訪問(wèn)pytorch查詢所需pytorch版本對(duì)應(yīng)的cuda版本號(hào)。在上，點(diǎn)擊查看舊版本的pytorch。根據(jù)需求選擇pytorch版本，找到對(duì)應(yīng)cuda版本。cuda及cudnn 右鍵桌面，選擇nvidia面板，或nvidia驅(qū)動(dòng)（若無(wú)此選項(xiàng)）。查看nvidia驅(qū)動(dòng)支持的cuda版本。