CUDA版本是 9.1.83,python版本3.6.5 win10
使用如下語句可以安裝cu91版本的pytorch
pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
然而下載torchvision會使用之前緩存中的torchvision而且0.4.1版本的並不能使用
最后使用
pip install torchvision==0.1.8
成功安裝
選擇對應的版本:
進入網頁選擇合適的pytorch版本:
https://download.pytorch.org/whl/torch_stable.html