torch.xpu.mem_get_info

torch.xpu.mem_get_info(device=None)

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

参数:

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

返回值:

设备上可用的内存，以字节为单位。int: 设备上的总内存，以字节为单位

返回类型:

int