PyTorch 与 CUDA 版本兼容表
感谢大家的分享,经测试可以训练推理的配置
| 硬件 | 操作系统 | CUDA版本 | Pytorch版本 |
|---|---|---|---|
| Orin Nano 8GB | Ubuntu-22.04 | 12.6 | 2.8.0 |
| Orin NX 16GB | Ubuntu-22.04 | 12.6 | 2.8.0 |
| RTX-4080 | Ubuntu-24.04 | 12.6 | 2.8.0 |
| RTX-5090D | Ubuntu | 12.8 | 2.7.1 |
感谢大家的分享,经测试可以训练推理的配置
| 硬件 | 操作系统 | CUDA版本 | Pytorch版本 |
|---|---|---|---|
| Orin Nano 8GB | Ubuntu-22.04 | 12.6 | 2.8.0 |
| Orin NX 16GB | Ubuntu-22.04 | 12.6 | 2.8.0 |
| RTX-4080 | Ubuntu-24.04 | 12.6 | 2.8.0 |
| RTX-5090D | Ubuntu | 12.8 | 2.7.1 |