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