温馨提示×

PyTorch在Ubuntu上支持哪些GPU

小樊
63
2025-08-30 00:51:24
栏目: 智能运维

PyTorch在Ubuntu上主要支持NVIDIA GPU,具体型号及注意事项如下:

  • GeForce系列:GTX 1050 Ti及以上(如RTX 20/30/40系列)。
  • Quadro系列:多数型号支持,具体需匹配CUDA版本。
  • Tesla系列:如T4、P4、V100、A100等,需对应CUDA版本。
  • 关键要求
    • 需安装与GPU兼容的CUDA Toolkit(如RTX 40系列需CUDA 11.8+)。
    • 需通过NVIDIA驱动支持,可通过nvidia-smi验证驱动和CUDA是否正常。
  • 版本匹配:PyTorch官网提供CUDA与PyTorch版本的兼容性表,需按需选择安装命令(如pipconda指定cudatoolkit版本)。

注:具体支持情况可通过PyTorch官方文档或torch.cuda.is_available()验证。

0