torch.cuda.memory.memory_snapshot

torch.cuda.memory.memory_snapshot(mempool_id=None, include_traces=True) [исходный код]

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

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

Параметры:

• mempool_id – Необязательный идентификатор пула памяти, для которого нужно получить снимок
• include_traces – Нужно ли включать записи трассировки в снимок. Если True (значение по умолчанию), включаются все записи трассировки. Если False, записи трассировки не включаются (облегчённый/быстрый снимок).

Примечание

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