温馨提示×

Debian下PyTorch支持哪些GPU

小樊
49
2025-08-12 08:45:47
栏目: 智能运维

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

  • GeForce系列:GTX 1050 Ti及以上型号,如GTX 1060、GTX 1070等;RTX 20系列,如RTX 2060、RTX 2070等;RTX 30系列,如RTX 3060、RTX 3070等;RTX 40系列,如RTX 4080、RTX 4090等。
  • Quadro系列:大多数Quadro GPU都受到支持,但具体型号可能因驱动版本而异。
  • Tesla系列:Tesla V100、P40、K80等型号被广泛支持。

同时,还需注意不同版本的PyTorch对CUDA版本有不同要求,需根据显卡型号选择合适的CUDA和PyTorch版本进行安装。

0