torch.cuda.memory_snapshot

torch.cuda.memory_snapshot() [source]

Возвращает снимок состояния выделения памяти CUDA на всех устройствах.

Интерпретация вывода этой функции требует знакомства с внутренним устройством выделения памяти.

Примечание

См. Управление памятью для получения дополнительной информации об управлении памятью GPU.