torch.accelerator.memory.max_memory_reserved#

torch.accelerator.memory.max_memory_reserved(device_index=None,/)[source]#

Return the currentaccelerator maximum device memory managed by the caching allocatorin bytes for a given device index.

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

Parameters

device_index (torch.device, str, int, optional) – the index of the device to target.If not given, usetorch.accelerator.current_device_index() by default.If atorch.device or str is provided, its type must match the currentaccelerator device type.

Return type

int