- 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 |
- cuda 与 cudnn关系
| cuda版本 | 支持cudnn版本 |
|---|---|
| cuda10.2 | v7.6.5 ~ v8.7.0 |
| cuda11.3 | v8.2.0 ~ v8.9.6 |
- 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 |
