怎么查看GPU架构

在Linux或macOS系统中，在终端中运行以下命令：

lspci | grep -i nvidia
您将看到一个输出，其中包含有关您的NVIDIA GPU的信息，包括其架构。例如，输出可能如下所示：

01:00.0 VGA compatible controller: NVIDIA Corporation GP104 [GeForce GTX 1080] (rev a1)
在这个例子中，GPU架构是 GP104。

在Windows系统中，您可以在任务管理器的性能标签页中找到有关NVIDIA GPU的信息，包括其型号和架构。您可以按照以下步骤查找该信息：

打开任务管理器：右键点击任务栏空白处，选择“任务管理器”。
切换到“性能”标签页。
在左侧栏中选择“GPU 0”（如果您的系统中有多个GPU，可以选择相应的GPU）。
在右侧窗口中，您将看到有关您的NVIDIA GPU的信息，包括架构。例如，它可能显示为“大于等于Pascal（SM 6.0）”，其中Pascal是GPU架构，SM 6.0是CUDA计算能力。

根据输出，您的GPU型号为"NVIDIA Corporation Device 28a1"，但是没有显示具体的架构。在这种情况下，您可以使用以下命令来获取更详细的信息：

nvidia-smi -q
这将显示有关您的NVIDIA GPU的详细信息，包括其型号、驱动程序版本和架构。您可以查找"Architecture"字段来查看GPU架构。例如，如果您的输出中包含以下行：

Architecture: 75
那么您的GPU架构为75，是Turing（图灵）架构的。

如果您的GPU架构是Ada Lovelace架构，您需要更新Makefile中的ARCH变量来支持该架构。更具体地说，您需要将-gencode arch=compute_35,code=sm_35更新为 -gencode arch=compute_86,code=[sm_86,compute_86]。

也就是说，您需要将包含ARCH的行：

ARCH= -gencode arch=compute_35,code=sm_35
修改为：

ARCH= -gencode arch=compute_86,code=[sm_86,compute_86]
这将确保您的代码可以在Ada Lovelace架构上编译和运行。在Makefile中进行更改后，您需要重新运行make命令来构建您的代码。