Ubuntu中PyTorch支持哪些GPU

在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等）。