本文前提是在已安装annaconda的前提下进行!
1.首先进入我们主菜单中的anaconda prompt终端,如下图所示:
2、进入prompt后,通过指令 conda create -n pp (环境名称,这个自己随便取)python=3.7
完成后,通过 conda activate pp(环境名) 来激活环境:
如下图所示:
3、通过pip install torch==1.11.0+cu115 torchvision==0.12.0+cu115 torchaudio==0.11.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html 安装cuda11.5对应的torch(gpu版本)
如下图所示:
4、Torch安装完成,我们通过python来验证torch是否安装成功,且是否可以使用gpu
>>> import torch
>>> torch.cuda.is_available()
如果结果显示为True,说明torch安装成功,且为gpu版本。
如果结果为False,则说明安装的torch为cpu版本。
如下图所示:
这里显示为True,则顺利完成torch(gpu版本)的安装!