1.首先激活虛擬環(huán)境

win+R+cmd 打開(kāi)dos命令窗口

查看虛擬環(huán)境列表

conda env list

激活虛擬環(huán)境

2.將原來(lái)的pytorch_cpu版本換成gpu版本

注意：安裝gpu版本的pytorch時(shí)并不需要先卸載原來(lái)的cpu版本pytorch,安裝時(shí)會(huì)自己替換的

打開(kāi)pytorch官網(wǎng)查看以前版本

Previous PyTorch Versions | PyTorch

根據(jù)自己gpu所能承載的cuda版本來(lái)下載對(duì)應(yīng)的pytorch

查看本機(jī)gpu所能承載的Cuda版本：

nvidia-smi

我所能承載的最高版本是12.2，所以Cuda12.2以及12.2一下的版本我都可以下載安裝，我選擇的是Cuda11.3

如果這行命令無(wú)效說(shuō)明還沒(méi)有裝顯卡驅(qū)動(dòng)，要先去nvidia官網(wǎng)去裝一下顯卡驅(qū)動(dòng)。

剛剛激活虛擬環(huán)境之后直接輸入一下命令:

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

然后等待安裝成功即可

3.驗(yàn)證是否安裝成功

python 
import torch 
torch.cuda.is_available() 

如果結(jié)果為T(mén)rue，說(shuō)明GPU版本的pytorch就安裝成功了

如上圖所示，安裝GPU版本的pytorch成功。