torch.cuda.memory_usage

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

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

Параметры:

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

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

int

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