PyTorch、CUDA、Python 版本对应表
高版本 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 | 3.6–3.9 | ✅ | ✅ | ✅ | ❌ | ❌ | ❌ | ❌ | +cu111, +cu102, +cu101 |
| 1.7.1 | 3.6–3.8 | ✅ | ✅ | ❌ | ❌ | ❌ | ❌ | ❌ | +cu110, +cu102, +cu101 |
| 1.6.0 | 3.6–3.8 | ✅ | ✅ | ❌ | ❌ | ❌ | ❌ | ❌ | +cu102, +cu101 |
| 1.5.1 | 3.5–3.8 | ✅ | ✅ | ❌ | ❌ | ❌ | ❌ | ❌ | +cu102, +cu101 |
| 1.4.0 | 3.5–3.8 | ✅ | ✅ | ❌ | ❌ | ❌ | ❌ | ❌ | +cu101, +cu100 |
PyTorch、CUDA、MMCV 版本组合
详细安装信息请参考 MMCV 官方文档。
MMCV-full 与 MMDetection 版本组合
| MMDetection 版本 | MMCV 版本 |
|---|---|
| master | mmcv-full>=1.3.17, <1.5.0 |
| 2.19.1 | mmcv-full>=1.3.17, <1.5.0 |
| 2.19.0 | mmcv-full>=1.3.17, <1.5.0 |
| 2.18.0 | mmcv-full>=1.3.17, <1.4.0 |
| 2.17.0 | mmcv-full>=1.3.14, <1.4.0 |
| 2.16.0 | mmcv-full>=1.3.8, <1.4.0 |
| 2.15.1 | mmcv-full>=1.3.8, <1.4.0 |
| 2.15.0 | mmcv-full>=1.3.8, <1.4.0 |
| 2.14.0 | mmcv-full>=1.3.8, <1.4.0 |
| 2.13.0 | mmcv-full>=1.3.3, <1.4.0 |
| 2.12.0 | mmcv-full>=1.3.3, <1.4.0 |
| 2.11.0 | mmcv-full>=1.2.4, <1.4.0 |
| 2.10.0 | mmcv-full>=1.2.4, <1.4.0 |
| 2.9.0 | mmcv-full>=1.2.4, <1.4.0 |
| 2.8.0 | mmcv-full>=1.2.4, <1.4.0 |
| 2.7.0 | mmcv-full>=1.1.5, <1.4.0 |
| 2.6.0 | mmcv-full>=1.1.5, <1.4.0 |
| 2.5.0 | mmcv-full>=1.1.5, <1.4.0 |
| 2.4.0 | mmcv-full>=1.1.1, <1.4.0 |
| 2.3.0 | mmcv-full==1.0.5 |
| 2.3.0rc0 | mmcv-full>=1.0.2 |
| 2.2.1 | mmcv==0.6.2 |
| 2.2.0 | mmcv==0.6.2 |
| 2.1.0 | mmcv>=0.5.9, <=0.6.1 |
| 2.0.0 | mmcv>=0.5.1, <=0.5.8 |
综合推荐配置
| MMDetection 版本 | MMCV-full 版本 | MMEngine 版本 | PyTorch 推荐版本 | 支持 CUDA 版本 | 架构说明 |
|---|---|---|---|---|---|
| 3.3.0 | ≥2.1.0 | ≥0.10.0 | 1.10 ~ 2.2 | 11.3 ~ 12.x | 全新 MMEngine 架构 |
| 3.2.x | 2.0.1 ~ 2.0.2 | ≥0.9.0 | 1.10 ~ 2.1 | 11.3 ~ 11.8 | 新架构适配中 |
| 3.0.0 | 2.0.0 | ≥0.7.1 | 1.10 ~ 2.0 | 11.1 ~ 11.7 | MMEngine 初期版本 |
| 2.28.2 | 1.7.0 | ❌ 不需要 | 1.10 ~ 1.13 | 10.2 ~ 11.6 | 经典架构最后版本 |
| 2.26.0 | 1.6.2 | ❌ | 1.8 ~ 1.12 | 10.1 ~ 11.3 | 适合向前兼容 |
| 2.24.1 | 1.5.2 | ❌ | 1.8 ~ 1.10 | 10.1 ~ 11.1 | 适合旧项目迁移 |
| 2.20.0 | 1.4.8 | ❌ | 1.7 ~ 1.10 | 10.0 ~ 10.2 | 适合 CUDA 10.1 |
| 2.18.1 | 1.3.18 | ❌ | 1.6 ~ 1.9 | 10.0 ~ 10.2 | VisDrone 项目常用 |
| 2.11.0 | 1.2.6 | ❌ | 1.6 ~ 1.8 | 10.0 ~ 10.1 | 稳定性不错 |
| 2.5.0 | 1.1.5 | ❌ | 1.5 ~ 1.6 | 10.0 | 较早期版本 |
| 2.0.0 | 1.0.0 | ❌ | 1.4 ~ 1.5 | 9.2 ~ 10.0 | V2 架构起点 |
| 1.2.0 | 0.6.2 | ❌ | 1.4 | 9.2 | MMDetection 早期版本 |
实际部署时，建议优先选择官方推荐的稳定版本组合。若需升级，务必先查阅对应版本的 Release Notes，避免引入破坏性变更。