巧妙利用NVIDIA-SMI查询GPU信息的细节和选项

2023-09-27 19:26:52

强制选项--format

--format选项用于指定查询结果的显示格式。它支持以下几种格式：

csv：以逗号分隔值（CSV）格式输出结果。
json：以JavaScript对象表示法（JSON）格式输出结果。
table：以表格格式输出结果。这是默认格式。

例如，要以CSV格式输出所有GPU的名称和利用率，可以使用以下命令：

nvidia-smi --query-gpu=name,utilization.gpu --format=csv

输出结果如下：

name,utilization.gpu
GPU-0,30%
GPU-1,60%
GPU-2,10%

附加选项-i、-f、-l和-lms

除了--format选项外，nvidia-smi --query-gpu还支持以下附加选项：

-i：指定要查询的GPU的索引。例如，要查询GPU-1的信息，可以使用以下命令：

nvidia-smi --query-gpu=name,utilization.gpu --i=1

-f：指定要查询的字段。例如，要查询GPU-0和GPU-1的名称、利用率和温度，可以使用以下命令：

nvidia-smi --query-gpu=name,utilization.gpu,temperature.gpu --f=0,1

-l：指定要查询的GPU的层级。例如，要查询所有GPU的名称和利用率，包括子GPU的信息，可以使用以下命令：

nvidia-smi --query-gpu=name,utilization.gpu --l=0,1

-lms：指定要查询的GPU的链接管理和电源状态。例如，要查询所有GPU的名称、链接状态和电源状态，可以使用以下命令：

nvidia-smi --query-gpu=name,pstate,power.draw --lms

组合使用选项

这些选项可以组合使用，以满足不同的查询需求。例如，要以JSON格式输出所有GPU的名称、利用率和温度，可以使用以下命令：

nvidia-smi --query-gpu=name,utilization.gpu,temperature.gpu --format=json

输出结果如下：

{
  "gpu_0": {
    "name": "GPU-0",
    "utilization.gpu": "30%",
    "temperature.gpu": "60C"
  },
  "gpu_1": {
    "name": "GPU-1",
    "utilization.gpu": "60%",
    "temperature.gpu": "70C"
  },
  "gpu_2": {
    "name": "GPU-2",
    "utilization.gpu": "10%",
    "temperature.gpu": "50C"
  }
}