API symbolscuda
Copy
tensorplay.cuda.memory_summary
- tensorplay.cuda.memory_summary(device: Any | None = None, abbreviated: bool = False) str[source]
Return a human-readable printout of the current memory allocator statistics for a given device.
This can be useful to display periodically during training, or when handling out-of-memory exceptions.
- Parameters:
device (tensorplay.Device or int, optional) – selected device. Returns printout for the current device, given by
current_device(), ifdeviceisNone(default).abbreviated (bool, optional) – whether to return an abbreviated summary (default: False).
Help improve this page
Found an error, an unclear step, or a missing example?
Was this page helpful?