torch.cuda.device_memory_used

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

Возвращает объем используемой глобальной памяти устройства в байтах согласно данным nvidia-smi или amd-smi.

Параметры:

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

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

int