torch.cuda.memory.reset_accumulated_memory_stats

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

Сбросить «накопленную» (историческую) статистику, отслеживаемую распределителем памяти CUDA.

Подробнее см. в memory_stats(). Накопленная статистика соответствует ключам “allocated” и “freed” в каждом отдельном словаре статистики, а также “num_alloc_retries” и “num_ooms”.

Параметры:

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

Примечание

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