torch.cuda.memory_summary

torch.cuda.memory_summary(device=None, abbreviated=False) [source]

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

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

Параметры:

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

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

str

Примечание

См. Управление памятью для получения дополнительной информации об управлении памятью GPU.