torch.cuda.max_memory_allocated
torch.cuda.max_memory_allocated
Определение
Функция PyTorch, возвращающая пиковое значение выделенной GPU-памяти в байтах с момента запуска процесса. Используется для оценки максимального потребления памяти при обучении или инференсе.