Python、PyTorch、CUDA 版本对应表
高版本 CUDA 可兼容低版本
| PyTorch 版本 | Python 支持版本 | CUDA 10.1 | CUDA 10.2 | CUDA 11.1 | CUDA 11.3 | CUDA 11.6 | CUDA 11.7 | CUDA 11.8 | 安装指令中 CUDA tag |
|---|---|---|---|---|---|---|---|---|---|
| 2.1.0 | 3.8–3.11 | ❌ | ❌ | ❌ | ❌ | ✅ | ✅ | ✅ | +cu117, +cu118 |
| 2.0.1 | 3.8–3.11 | ❌ | ❌ | ❌ | ❌ | ✅ | ✅ | ✅ | +cu117, +cu118 |
| 1.13.1 | 3.7–3.10 | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | +cu117, +cu116, etc. |
| 1.12.1 | 3.7–3.10 | ❌ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | +cu116, +cu115, etc. |
| 1.11.0 | 3.6–3.9 | ❌ | ✅ | ✅ | ✅ | ✅ | ❌ | ❌ | +cu113, +cu102 |
| 1.10.0 | 3.6–3.9 | ✅ | ✅ | ✅ | ✅ | ❌ | ❌ | ❌ | +cu113, +cu102, +cu101 |
| 1.9.0 | 3.6–3.9 | ✅ | ✅ | ✅ | ❌ | ❌ | ❌ | ❌ | +cu111, +cu102, +cu101 |
| 1.8.0 |


