torch.cuda.memory.memory_summary

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

Возвращает удобочитаемую распечатку текущей статистики распределителя памяти для указанного устройства.

Это может быть полезно для периодического отображения во время обучения или при обработке исключений, связанных с нехваткой памяти.

Параметры:

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

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

str

Примечание

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