tensorplay.cuda.memory_snapshot

tensorplay.cuda.memory_snapshot(mempool_id=None, include_traces=True)[source]

Return a snapshot of the CUDA memory allocator state across all devices.

Interpreting the output of this function requires familiarity with the memory allocator internals. Not exposed by this TensorPlay build.