关于python版本,显卡版本,torch版本之间的问题
第一部分:显卡架构(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 最佳支持 |