cpytorch-wheels-cu121安装包下载_开源镜像站-阿里云 (aliyun.com)https://mirrors.aliyun.com/pytorch-wheels/cu121/从上述网址中找torch的包,比直接从官网安装快很多
我的cuda是12.1,用的是python3.10,torch版本根据自己想要的安装,但是torch,torchaudio ,torchvision版本要对应,比如下面cu121对应的是CUDA12.1,cp310-win_amd是对应的windows下的python3.10。
下载完成后cd到上述文件的目录,3个文件直接使用pip进行安装,例如
pip install "torchvision-0.18.1+cu121-cp310-cp310-win_amd64.whl"
3个文件都安装完毕后在终端输入
>>> import torch
>>> print(torch.__version__)
输出应该是2.3.1+cu121,这和自己下载的版本是对应的
最后输入下列命令
print(torch.cuda.is_available())
如果输出为True则证明安装成功,可以使用torch的GPU版本了