torch.cuda.memory.memory_reserved

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

Возвращает текущий объем памяти GPU в байтах, управляемый кэширующим распределителем для указанного устройства.

Параметры:

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

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

int

Примечание

Подробнее об управлении памятью GPU см. в разделе Управление памятью.