torch.cuda.max_memory_allocated

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

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

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

Параметры

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

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

int

Примечание

Дополнительные сведения о управлении памятью GPU см. в Управлении памятью.