API symbolscuda
Copy
tensorplay.cuda.memory_allocated
- tensorplay.cuda.memory_allocated(device: Any | None = None) int[source]
Return the current GPU memory occupied by tensors in bytes for a given device.
- Parameters:
device (tensorplay.Device or int, optional) – selected device. Returns statistic for the current device, given by
current_device(), ifdeviceisNone(default).
Help improve this page
Found an error, an unclear step, or a missing example?
Was this page helpful?