TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.reset_max_memory_allocated

tensorplay.cuda.reset_max_memory_allocated(device: Any | None = None) → None[source]

Reset the starting point in tracking maximum GPU memory occupied by tensors for a given device.

Warning

This function now calls reset_peak_memory_stats(), which resets /all/ peak memory stats.

tensorplay.cuda.reset_accumulated_memory_stats

Reset the “accumulated” (historical) stats tracked by the CUDA memory allocator.

tensorplay.cuda.reset_max_memory_cached

Reset the starting point in tracking maximum GPU memory managed by the caching allocator for a given device.

Ask DeepWiki