【PyTorch安裝】關於 PyTorch, torchvision 和 CUDA 版本的對應關系

一直以來對於軟件的版本對應關系有困惑,其實我們可以從這個官方鏈接上得到指點:

https://download.pytorch.org/whl/torch_stable.html

比如我們要安裝 PyTorch1.4.0,可以先從上面網站上找到對應關系,再使用以下命令進行下載:

pip install torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

---

更新:
如果安裝 cu90 對應的 torch,可以參考如下網址:

https://download.pytorch.org/whl/cu90/torch_stable.html

---

PyTorch 歷史版本的下載:

https://pytorch.org/get-started/previous-versions/