顺德品牌网站建设汉阳网站建设公司

1.首先激活虚拟环境

win+R+cmd 打开dos命令窗口

查看虚拟环境列表

conda env list

激活虚拟环境

2.将原来的pytorch_cpu版本换成gpu版本

注意：安装gpu版本的pytorch时并不需要先卸载原来的cpu版本pytorch,安装时会自己替换的

打开pytorch官网查看以前版本

Previous PyTorch Versions | PyTorch

根据自己gpu所能承载的cuda版本来下载对应的pytorch

查看本机gpu所能承载的Cuda版本：

nvidia-smi

我所能承载的最高版本是12.2，所以Cuda12.2以及12.2一下的版本我都可以下载安装，我选择的是Cuda11.3

如果这行命令无效说明还没有装显卡驱动，要先去nvidia官网去装一下显卡驱动。

刚刚激活虚拟环境之后直接输入一下命令:

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.验证是否安装成功

python 
import torch 
torch.cuda.is_available() 

如果结果为True，说明GPU版本的pytorch就安装成功了

如上图所示，安装GPU版本的pytorch成功。