torch.xpu.mem_get_info¶
- torch.xpu.mem_get_info(device=None)[來源][來源]¶
傳回指定裝置的全域可用和總 GPU 記憶體。
- 參數
device (torch.device 或 int 或 str, optional) – 選取的裝置。如果
device為None(預設)，則傳回由current_device()給定的目前裝置的統計資訊。- 傳回
以位元組為單位表示裝置上的可用記憶體。 int: 以位元組為單位表示裝置上的總記憶體
- 傳回類型
