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.