温馨提示×

Ubuntu中PyTorch支持哪些GPU

小樊
51
2025-09-15 01:37:41
栏目: 智能运维

在Ubuntu中,PyTorch主要支持NVIDIA的GPU,具体如下:

  • 数据中心GPU系列:如L系列(L40、L40S等)、H系列(H100、H800等)、A系列(A100、A40等)、RTX系列(RTX 5000 Ada Generation等)、Tesla系列(Tesla V100、P100等)。
  • 消费级GPU系列:主要是GeForce系列的部分型号,如RTX 30系列(RTX 3060、RTX 3070等)、RTX 40系列(RTX 4090、RTX 4080等)。

0