torch.cuda.memory_snapshot

torch.cuda.memory_snapshot() [source]

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

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

Примечание

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