知识库/参考文档/PyTorch 与 CUDA 版本兼容表

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