.
背景
在使用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)也列在其中。
以上。
希望能够帮到大家,同时如果有更简易更精简的输出,欢迎大家讨论区讨论!
.
题后记
另外,祝大家平安夜快乐!圣诞快乐!
转眼又是一年圣诞节,今年的心情还是有些复杂的。我知道还需要时间,也有可能是自己的逃避。但不管怎么样,圣诞节总是令人快乐和温暖的 ^ ________________________________ ^!
接下来,继续加油!