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

以上就是关于“PyTorch在Linux上支持哪些GPU”的相关介绍，筋斗云是国内较早的云主机应用的服务商，拥有10余年行业经验，提供丰富的云服务器、租用服务器等相关产品服务。云服务器资源弹性伸缩，主机vCPU、内存性能强悍、超高I/O速度、故障秒级恢复；电子化备案，提交快速，专业团队7×24小时服务支持！

简单好用、高性价比云服务器租用链接：https://www.jindouyun.cn/product/cvm