在学习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,大功告成。