cuda_memory_snapshot: Capture CUDA Memory State Snapshot

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Capture CUDA Memory State Snapshot

Description

Saves a snapshot of the CUDA memory state at the time it was called. The resulting binary output is in pickle format and can be visualized using the interactive snapshot viewer available at pytorch.org/memory_viz.

Usage

cuda_memory_snapshot()

Value

Raw binary data representing the snapshot in pickle format.

Examples

if (torch_is_installed()) {
## Not run: 
snapshot <- cuda_memory_snapshot()

## End(Not run)
}

torch documentation built on Aug. 21, 2025, 5:50 p.m.