torch、torchvision、cuda 、python对应版本匹配

torchtorchvisioncudapython
1.8.00.8.210.2,11.0>=3.6
1.7.10.8.29.2 ,10.1 ,10.2 ,11.0>=3.6
1.7.00.8.19.2 ,10.1, 10.2 ,11.0>=3.6
1.7.00.8.09.2 ,10.1 ,10.2 ,11.0>=3.6
1.6.00.7.09.2, 10.1 ,10.2>=3.6
1.5.10.6.19.2, 10.1 ,10.2>=3.5
1.5.00.6.09.2 ,10.1, 10.2>=3.5
1.4.00.5.09.2, 10.1==2.7, >=3.5, <=3.8
1.3.10.4.29.2 ,10.1==2.7, >=3.5, <=3.7
1.3.00.4.19.2 ,10.1==2.7, >=3.5, <=3.7
1.2.00.4.09.2 ,10.0==2.7, >=3.5, <=3.7
1.1.00.3.09.0, 10.0==2.7, >=3.5, <=3.7
1.0.10.2.29.0 ,10.0==2.7, >=3.5, <=3.7
1.0.00.2.28.0 ,9.0 ,10.0==2.7, >=3.5, <=3.7
0.4.10.2.28.0 ,9.0 ,10.0==2.7, >=3.5, <=3.7

pip install torch == 1.3.0 torchvision ==0.4.1 默认相应cuda最高版本10.1
参考:https://pytorch.org/get-started/previous-versions/

Logo

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

更多推荐