题主在用 conda 和 pip 下载 torch 时速度相当慢(清华源也很慢),一般网速好的时候也只停留在 200~300KB/S,这对于pytorch的大小动不动就超过1~2G来说是相当折磨人的,往往要等上接近一小时才能下好,故可以采用手动下载和安装的方式,能节省相当多的时间,废话不多说,引入正题:
1. 先进入官网Previous PyTorch Versions | PyTorch 找到自己要下载的torch版本,以torch2.0.0+cu118版本为例:
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
2. 进入pytorch的阿里源官网pytorch-wheels安装包下载_开源镜像站-阿里云 (aliyun.com),往下拉找到 cuda11.8 (cu118) 对应的下载链接pytorch-wheels-cu118安装包下载_开源镜像站-阿里云 (aliyun.com),点击进去:
3. 下拉找到自己要下载的版本,以python3.9和cuda11.8对应的 torch 2.0.0 版本为例,linux的名称为torch-2.0.0+cu118-cp39-cp39-linux_x86_64.whl,torch-2.0.0指 torch版本号,cp39指python3.9,linux版本名称带有linux;同理,windows的版本名称为 torch-2.0.0+cu118-cp39-cp39-win_amd64.whl:
4. 点击对应名称选项即可下载,下载速度如下:
5. 下载完成后进入虚拟环境,并 cd 至torch的下载目录,使用 pip install + 文件名进行安装:
6. 安装成功后再使用第一步中pytorch官网获取的命令完成对应版本的torchvision和torchaudio的安装,因为这两个库的文件较小,不用阿里源也可以较快安装:
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
命令会自动识别虚拟环境内有无安装torch,若已经安装,则不会重复下载安装torch,继续下载torchvision和torchaudio: