本文前提是在已安装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版本)的安装!

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐