温馨提示×

Ubuntu中PyTorch支持哪些GPU

ubuntu

小樊

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 赞

0 踩