torch.xpu.mem_get_info — PyTorch 2.7 documentation (original) (raw)

torch.xpu.mem_get_info(device=None)[source][source]¶

Return the global free and total GPU memory for a given device.

Parameters

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

Returns

the memory available on the device in units of bytes. int: the total memory on the device in units of bytes

Return type

int

• torch.xpu.mem_get_info
  • mem_get_info()

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.