第一部分:显卡架构(GPU Compute Capability)对照表
| 显卡系列 | 架构 | Compute Capability | 是否被 PyTorch 支持 |
|---|---|---|---|
| GTX 10 系 | Pascal | sm_61 | ✔ 全部支持 |
| RTX 20 系 | Turing | sm_75 | ✔ 全部支持 |
| RTX 30 系 | Ampere | sm_80 / sm_86 | ✔ 全部支持 |
| RTX 40 系 | Ada | sm_89 | ✔ PyTorch ≥ 2.0 支持 |
| RTX 50 系(5080/5090) | Blackwell | sm_120 | ✔ PyTorch ≥ 2.5 才支持 |
| Tesla A100 | Ampere | sm_80 | ✔ |
| H100 | Hopper | sm_90 | ✔ |
| H200 / GB200 | Blackwell | sm_120 / sm_130 | ✔ PyTorch ≥ 2.6 |
第二部分:Python ↔ PyTorch 支持矩阵(官方)
| PyTorch 版本 | 支持的 Python | 支持的 CUDA | RTX 5080 支持? |
|---|---|---|---|
| 1.10 | 3.6--3.9 | 10.2 / 11.3 | ❌ 完全不支持 |
| 1.12 | 3.7--3.10 | 11.3 / 11.6 | ❌ 不支持 |
| 2.0 | 3.8--3.11 | 11.7 / 11.8 | ❌ 不支持 |
| 2.1 | 3.8--3.11 | 12.1 | ❌ 不支持 |
| 2.2 | 3.8--3.12 | 12.1 | ❌ 不支持 |
| 2.3 | 3.8--3.12 | 12.1 | ❌ 不支持 |
| 2.4 | 3.8--3.12 | 12.1 | ❌ 不支持 |
| 2.5 | 3.10--3.12 | 12.4 | ✔ 实验性支持 SM120 |
| 2.6 | 3.10--3.12 | 12.5 | ✔ 稳定支持 SM120 |
| 2.7 | 3.10--3.12 | 12.6 | ✔ 完整支持 SM120 |
| Nightly (2.8+) | 3.11--3.12 | 12.7 / 12.8 | ✔ Blackwell 最佳支持 |