tensorplay.cuda.max_memory_allocated

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

Return the maximum GPU memory occupied by tensors in bytes for a given device.

By default, this returns the peak allocated memory since the beginning of this program. reset_peak_memory_stats() can be used to reset the starting point in tracking this metric.

Parameters:

device (tensorplay.Device or int, optional) – selected device. Returns statistic for the current device, given by current_device(), if device is None (default).