pytorch与cuda版本对应关系汇总

1. pytorch与cuda版本关系

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

2. cuda 与 cudnn关系

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

4. pytorch 与 python关系

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