cuda11.5对应的pytorch(gpu版本)安装教程,手把手教你一步到位!

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值