温馨提示×

Ubuntu支持PyTorch哪些版本

小樊
81
2025-08-09 19:06:48
栏目: 智能运维

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的版本之一。

0