torch.xpu.memory.mem_get_info

torch.xpu.memory.mem_get_info(device=None)

返回给定设备的全局空闲内存和总 GPU 内存。

参数:

device (torch.device 或 int 或 str, 可选) – 选定的设备。如果 device 为 None（默认值），则返回当前设备（由 current_device() 提供）的统计信息。

返回:

一个包含两个整数的元组 (free_memory, total_memory)，单位为字节。

第一个值为设备上的空闲内存（所有进程和应用程序均可使用的内存），第二个值为设备的硬件内存总容量。

返回类型:

tuple[int, int]