torch.xpu.mem_get_info¶
- torch.xpu.mem_get_info(device=None)[source][source]¶
返回给定设备的全局空闲内存和总内存。
- 参数
device (torch.device 或 int 或 str，可选) – 所选设备。如果
device为None（默认），则返回当前设备（由current_device()给出）的统计信息。- 返回值
设备上可用的内存量（字节为单位）。int：设备上的总内存量（字节为单位）
- 返回值类型
返回给定设备的全局空闲内存和总内存。
device (torch.device 或 int 或 str，可选) – 所选设备。如果 device 为 None（默认），则返回当前设备（由 current_device() 给出）的统计信息。
设备上可用的内存量（字节为单位）。int：设备上的总内存量（字节为单位）
为了分析流量和优化您的体验，我们在此网站上使用 cookie。点击或导航即表示您同意我们使用 cookie。作为本网站当前的维护者，适用 Facebook 的 Cookie 政策。了解更多信息，包括有关可用控件的信息：Cookie 政策。
