torch.cuda.max_memory_reserved

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

Возвращает максимальный объём памяти на GPU, управляемый кэширующим выделенным блоком, в байтах для заданного устройства.

По умолчанию возвращает пиковый объём кэшированной памяти с начала работы программы. reset_peak_memory_stats() можно использовать для сброса точки отсчёта при отслеживании этой метрики. Например, эти две функции могут измерять пиковый объём кэшированной памяти на каждой итерации цикла обучения.

Параметры

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

Тип возвращаемого значения

int

Примечание

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