首先确定好版本:
在官网中选择版本,注意CUDA版本就行
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu121
在终端输入命令后点击下载torch的链接,在迅雷中新建任务,下载到本地复制到项目中
然后在终端输入命令从本地安装:
pip install "torch-2.1.1+cu121-cp38-cp38-win_amd64.whl"
安装完毕后再次运行上面命令,安装torchvison,就可以了:
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu121
标签:cu121,本地,whl,torch,install,pip,2.1,安装
From: https://blog.csdn.net/weixin_43200362/article/details/144545067