PyTorch官网不支持conda,使用pip下载的方法

在学习PyTorch时,来官网复制安装,结果发现NOTE: Conda packages are no longer available. Please use pip instead.(Conda包不再可用。请使用pip。)

在这里插入图片描述

在保证不发生错误的情况于是查找如何使用pip命令。

先创建一个自己的虚拟环境:

conda create -n pytorch python=3.8

切换环境

conda activate pytorch

在anaconda下用pip装包的原因:**官方发布公告说conda维护成本过高而不在维护。**尽管在anaconda下我们可以很方便的使用conda install来安装我们需要的依赖,但是anaconda本身只提供部分包,远没有pip提供的包多,有时conda无法安装我们需要的包,我们需要用pip将其装到conda环境里。

首先,我们需要判断目前我们用的pip指令,会把包装到哪里,通常情况下,pip不像conda一样,他不知道环境,我们首先要确保我们用的是本环境的pip,这样pip install时,包才会创建到本环境中,不然包会创建到base环境,供各个不同的其他conda环境共享,此时可能会产生版本冲突问题(不同环境中可能对同一个包的版本要求不同)

通过where pip 看 是否在当前虚拟环境下,新版本一般是有的。

where pip

若环境内没有,则安装pip

conda install pip

通过conda list,可以检查环境中的包

conda list

之后就可以通过pip来下载PyTorch了,我下载的是11.8版本

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

或者试一试原来的conda版本可不可行(我没有试过)

conda install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 pytorch-cuda=12.1 -c pytorch -c nvidia

如果嫌弃慢可以使用镜像(也不快)

pip3 install torch==2.4.1 torchvision torchaudio -f https://mirrors.aliyun.com/pytorch-wheels/cu121/

再或者把文件下到本地,我放在了C:\Users\sd\Downloads(我用的这个方法,不过和人家视频上面的教学稍有不一样,不过结果)

(base) C:\Users\sd>conda activate torch37
(torch37) C:\Users\sd>cd C:\Users\sd\Downloads
(torch37) C:\Users\sd\Downloads>pip install "torch-2.3.1+cu121-cp38-cp38-win_amd64.whl"

进入python模式,输入import torch没有出错,代表安装完成。

(pytorch) C:\Users\sd>python
Python 3.8.20 (default, Oct  3 2024, 15:19:54) [MSC v.1929 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'2.3.1+cu121'
>>>

输入torch.cuda.is_available()检测GPU是否可用

torch.cuda.is_available()

若为True,则可以使用GPU,大功告成。

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值