cuda_memory_snapshot {torch} | R Documentation |
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)
}
[Package torch version 0.15.1 Index]