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 см. Управление памятью.