Ubuntu支持多个PyTorch版本,具体取决于CUDA版本,以下是部分常见版本:
- PyTorch 2.7.x:支持CUDA 12.8,适用于NVIDIA 50/40系显卡,如RTX 5090/4080。
- PyTorch 2.6.x:支持CUDA 12.6,适用于NVIDIA 30系显卡,如RTX 3090/3080。
- PyTorch 2.5.x:支持CUDA 11.8/12.1,可兼容旧设备,如GTX 16/10系显卡。
- PyTorch 2.4.x:需CUDA 12.1/12.4。
- PyTorch 2.3.x:最后支持CUDA 11.8的版本之一。