中国专门做统计网站,苏州培训网站建设,全省建设信息网站,h5网站建设1.首先激活虚拟环境
winRcmd 打开dos命令窗口 查看虚拟环境列表
conda env list 激活虚拟环境 2.将原来的pytorch_cpu版本换成gpu版本
注意#xff1a;安装gpu版本的pytorch时并不需要先卸载原来的cpu版本pytorch,安装时会自己替换的
打开pytorch官网查看以前版本
Previo…1.首先激活虚拟环境
winRcmd 打开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 torch1.12.1cu113 torchvision0.13.1cu113 torchaudio0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
然后等待安装成功即可
3.验证是否安装成功
python
import torch
torch.cuda.is_available()
如果结果为True说明GPU版本的pytorch就安装成功了 如上图所示安装GPU版本的pytorch成功。
