温馨提示×

PyTorch在Linux上支持哪些GPU

linux

小樊

58

2025-08-27 19:10:20

栏目: 智能运维

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版本。

注意：

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

0 赞

0 踩

最新问答

相关问答

相关标签

产品服务

地区划分

专题活动

帮助支持

关于我们

售后咨询

7*24小时在线电话：400-100-2938

7*24小时在线 QQ：800811969

关注亿速云

亿速云公众号

手机网站二维码