torch.xpu.device_memory_used

torch.xpu.device_memory_used(device=None)

Return the current GPU used global (device) memory in bytes.

Parameters:

device (torch.device, str or int, optional) – selected device. Uses the current device, given by current_device(), if None (default).

Return type:

int

Note

This API may require elevated privileges (e.g. sudo) to access GPU memory usage information.