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

分享最新的 NVIDIA AI Software 资源以及活动/会议信息,精选收录AI相关技术内容,欢迎大家加入社区并参与讨论。

更多推荐