安装PyTorch后，如何查询它支持哪些显卡型号？

安装 PyTorch 后如何快速查询它支持哪些显卡型号？

痛点：很多小白在完成 PyTorch 安装后面对海量的显卡型号与 CUDA 版本信息感到迷茫。不知道自己的显卡是否真的能被 PyTorch 利用，导致 GPU 加速无法正常工作。

一、先弄清楚自己到底有什么显卡

打开，页面右侧会自动检测并显示你的显卡信息。其实，如果没有自动检测，你可以手动在程序里查看：

• Windows：右键“此电脑” → “属性” → “设备管理器” → “显示适配器”。
• Linux：运行 lspci | grep -i nvidia或 lspci | grep -i vga。

二、官方文档——最权威的兼容列表

PyTorch 官方文档会列出每个 PyTorch 发行版对应的 CUDA 支持情况还有兼容的 NVIDIA GPU 型号。

1. 进入。
2. 选择你已安装的 PyTorch 版本（例如 torch==2.2.0)。
3. 在对应的 “CUDA Compatibility” 小节中，找到官方给出的 “Supported GPU list”。

三、命令行直接查询

痛点：不想打开网页，又想马上确认显卡是否被识别？按理说，下面的几行代码帮你立刻得到答案。