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 (по умолчанию).

Примечание

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