怎么查看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命令来构建您的代码。

你可能感兴趣的:(架构)