.

背景

在使用GPU进行AI模型训练或推理部署的时候,我们通常需要确认当前系统中的硬件信息,如GPU型号、GPU数量、GPU可用显存等等。

在NVIDIA GPU上,指令nvidia-smi默认打印以上所有的常用信息。对应来说,AMD GPU中常用指令rocm-smi,打印信息中目前(2024年12月24日平安夜)还不包含GPU型号。

待解决问题:如果希望查看当前系统中所配置的AMD GPU的型号,应该如何操作并将信息打印出来呢?

.

rocm-smi --showallinfo

针对这个问题,我们目前找到的最简单同时也是最繁琐的方法是使用–showallinfo参数,指令如下:

rocm-smi --showallinfo

使用该指令后,系统会输出所有和AMD GPU相关的信息。之所以称之为最繁琐的方法之一是因为该指令会把所有信息打印。在位于最前面的输出块,打印的信息会包含系统中所有GPU的ID信息,GPU型号(Device Name)也列在其中。

以上。

希望能够帮到大家,同时如果有更简易更精简的输出,欢迎大家讨论区讨论!

.

题后记

另外,祝大家平安夜快乐!圣诞快乐!

转眼又是一年圣诞节,今年的心情还是有些复杂的。我知道还需要时间,也有可能是自己的逃避。但不管怎么样,圣诞节总是令人快乐和温暖的 ^ ________________________________ ^!

接下来,继续加油!

.

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐