torch.cuda.utilization

torch.cuda.utilization(device=None) [исходный код]

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

Параметры:

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

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

int

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