更新了torch1.7.1对应的版本:

torchtorchvisionpython
master / nightlymaster / nightly>=3.6
1.7.10.8.2>=3.6
1.7.00.8.1>=3.6
1.7.00.8.0>=3.6
1.6.00.7.0>=3.6
1.5.10.6.1>=3.5
1.5.00.6.0>=3.5
1.4.00.5.0==2.7>=3.5<=3.8
1.3.10.4.2==2.7>=3.5<=3.7
1.3.00.4.1==2.7>=3.5<=3.7
1.2.00.4.0==2.7>=3.5<=3.7
1.1.00.3.0==2.7>=3.5<=3.7
<=1.0.10.2.2==2.7>=3.5<=3.7

命令:

pip install torchvision

在anaconda下:

conda install torchvision -c pytorch

下载地址:

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

 

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐