PyTorch在Linux上支持哪些GPU

PyTorch在Linux上主要支持NVIDIA GPU，具体型号如下：

• 消费级系列：GeForce GTX 1050 Ti及以上（如GTX 1060、RTX 2060、RTX 3060/70/80/90等）。
• 专业/数据中心系列：Quadro P/M/V系列、Tesla V100/P100/K80/T4等。
• 最新系列：部分RTX 40系（如RTX 4090）和H100等需匹配CUDA 12.x+的PyTorch版本。

注意：

1. 需安装与GPU兼容的CUDA Toolkit和驱动，可通过nvidia-smi查看驱动支持的CUDA版本。
2. PyTorch版本与CUDA版本需严格匹配，例如PyTorch 2.7+需CUDA 12.8+。
3. 安装时建议通过conda指定CUDA版本（如conda install pytorch cudatoolkit=12.8）。