torch.cuda.reset_max_memory_allocated

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

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

См. max_memory_allocated() для подробностей.

Параметры:

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

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

Эта функция теперь вызывает reset_peak_memory_stats(), что сбрасывает /все/ пиковые статистические данные памяти.

Примечание

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