pytorch轮子下载
pytorch轮子下载地址:https://download.pytorch.org/whl/cu113/torch_stable.html
以1.11的cuda版本为例:
需要下载:
torch-1.11.0+cu113-cp39-cp39-linux_x86_64.whl
torchaudio-0.11.0+cu113-cp39-cp39-linux_x86_64.whl
torchvision-0.12.0+cu113-cp39-cp39-linux_x86_64.whl
可以注意到torch和torchaudio都是11之类的编号,但是torchvision却用的12,这个不好分辨,建议去
pytoch官网找到那个安装命令,然后试一试,下载的时候看到真实用的链接,然后ctrlC掉然后可以自己下载
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113