torch.cuda.utilization

torch.cuda.utilization(device=None)

返回在过去采样周期内，GPU上有一个或多个内核在执行的时间百分比，如nvidia-smi所示。

Parameters

设备 (torch.device 或 int, 可选) – 选择的设备。返回当前设备的统计信息，由 current_device() 给出，如果 device 为 None（默认）。

Return type

int

警告：每个采样周期可能介于1秒到1/6秒之间，具体取决于所查询的产品。