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.