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.