torch.cuda.reset_peak_memory_stats

torch.cuda.reset_peak_memory_stats(device=None) [source]

Сбрасывает статистику «пиковых» значений, отслеживаемую выделением памяти CUDA.

См. memory_stats() для получения подробностей. Пиковые значения соответствуют ключу “peak” в каждом отдельном словаре статистики.

Параметры:

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

Примечание

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