torch.cuda.utilization

torch.cuda.utilization(device=None) [source]

Возвращает процент времени за последний период выборки, в течение которого один или несколько ядер выполнялись на GPU, как указано в nvidia-smi.

Параметры

device (torch.device или int, необязательно) – выбранное устройство. Возвращает статистику для текущего устройства, заданного функцией current_device(), если device равна None (по умолчанию).

Тип возвращаемого значения

int

Предупреждение: Каждый период выборки может составлять от 1 секунды до 1/6 секунды в зависимости от продукта, который запрашивается.