torch.cuda.memory.max_memory_allocated

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

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

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

Параметры:

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

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

int

Примечание

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