pytorch与cuda版本对应关系汇总_pytorch cuda版本对应关系

作者：小蓝xlanll | 2024-03-01 11:04:56

踩

pytorch cuda版本对应关系

                        
                    pytorch与cuda版本关系
cuda版本支持pytorch版本
cuda10.21.5 ~ 1.12
cuda11.01.7 ~ 1.7.1
cuda11.11.8 ~ 1.10.1
cuda11.31.8.1 ~ 1.12.1
cuda11.61.12.0 ~ 1.13.1
cuda11.71.13.0 ~ 2.0.1
cuda11.82.0.0 ~ 2.1.1
cuda12.12.1.0 ~ 2.1.1
cuda 与 cudnn关系
cuda版本支持cudnn版本
cuda10.2v7.6.5 ~ v8.7.0
cuda11.3v8.2.0 ~ v8.9.6
pytorch 与 python关系
torchtorchvisionPython
main / nightlymain / nightly>=3.8, <=3.11
2.00.15>=3.8, <=3.11
1.130.14>=3.7.2, <=3.10
1.120.13>=3.7, <=3.10
1.110.12>=3.7, <=3.10
1.100.11>=3.6, <=3.9
1.90.10>=3.6, <=3.9
1.80.9>=3.6, <=3.9
1.70.8>=3.6, <=3.9
1.60.7>=3.6, <=3.8
1.50.6>=3.5, <=3.8
1.40.5==2.7, >=3.5, <=3.8
1.30.4.2 / 0.4.3==2.7, >=3.5, <=3.7
1.20.4.1==2.7, >=3.5, <=3.7
1.10.3==2.7, >=3.5, <=3.7
<=1.00.2==2.7, >=3.5, <=3.7

cuda版本	支持pytorch版本
cuda10.2	1.5 ~ 1.12
cuda11.0	1.7 ~ 1.7.1
cuda11.1	1.8 ~ 1.10.1
cuda11.3	1.8.1 ~ 1.12.1
cuda11.6	1.12.0 ~ 1.13.1
cuda11.7	1.13.0 ~ 2.0.1
cuda11.8	2.0.0 ~ 2.1.1
cuda12.1	2.1.0 ~ 2.1.1

cuda版本	支持cudnn版本
cuda10.2	v7.6.5 ~ v8.7.0
cuda11.3	v8.2.0 ~ v8.9.6

torch	torchvision	Python
main / nightly	main / nightly	>=3.8, <=3.11
2.0	0.15	>=3.8, <=3.11
1.13	0.14	>=3.7.2, <=3.10
1.12	0.13	>=3.7, <=3.10
1.11	0.12	>=3.7, <=3.10
1.10	0.11	>=3.6, <=3.9
1.9	0.10	>=3.6, <=3.9
1.8	0.9	>=3.6, <=3.9
1.7	0.8	>=3.6, <=3.9
1.6	0.7	>=3.6, <=3.8
1.5	0.6	>=3.5, <=3.8
1.4	0.5	==2.7, >=3.5, <=3.8
1.3	0.4.2 / 0.4.3	==2.7, >=3.5, <=3.7
1.2	0.4.1	==2.7, >=3.5, <=3.7
1.1	0.3	==2.7, >=3.5, <=3.7
<=1.0	0.2	==2.7, >=3.5, <=3.7

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小蓝xlanll/article/detail/173395