torch.cuda.memory.reset_max_memory_allocated

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

Сбросить начальную точку отслеживания максимального объёма памяти GPU, занятой тензорами для указанного устройства.

Подробности см. в max_memory_allocated().

Параметры:

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

Предупреждение

Теперь эта функция вызывает reset_peak_memory_stats(), которая сбрасывает /всю/ статистику пикового использования памяти.

Примечание

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