torch.xpu.mem_get_info¶
- torch.xpu.mem_get_info(device=None)[source][source]¶
返回給定裝置的全域性空閒記憶體和總記憶體。
- 引數
device (torch.device 或 int 或 str，可選) – 所選裝置。如果
device為None（預設），則返回當前裝置（由current_device()給出）的統計資訊。- 返回值
裝置上可用的記憶體量（位元組為單位）。int：裝置上的總記憶體量（位元組為單位）
- 返回值型別
