torch.cuda.memory.reset_max_memory_cached

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

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

Подробнее см. max_memory_cached().

Параметры:

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

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

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

Примечание

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