torch.cuda.reset_max_memory_cached

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

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

См. max_memory_cached() для получения подробной информации.

Параметры:

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

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

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

Примечание

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