pytorch与cuda版本对应关系汇总

AICVer

已于 2023-12-28 10:34:03 修改

阅读量3.1w

点赞数 48

分类专栏：深度学习文章标签：深度学习

于 2023-12-27 16:56:19 首次发布

本文链接：https://blog.csdn.net/u011489887/article/details/135250561

版权

深度学习专栏收录该内容

61 篇文章 1 订阅

订阅专栏

                    
                        
                    
                    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