torch.xpu.memory_reserved¶
- torch.xpu.memory_reserved(device=None)[source][source]¶
返回給定裝置上由快取分配器管理的當前 GPU 記憶體量（以位元組為單位）。
- 引數
device (torch.device 或 int 或 str, 可選) – 所選裝置。如果
device為None(預設)，則返回由current_device()給出的當前裝置的統計資訊。- 返回型別
返回給定裝置上由快取分配器管理的當前 GPU 記憶體量（以位元組為單位）。
device (torch.device 或 int 或 str, 可選) – 所選裝置。如果 device 為 None (預設)，則返回由 current_device() 給出的當前裝置的統計資訊。
為了分析流量和最佳化您的體驗，本網站會使用 Cookie。點選或導航即表示您同意我們使用 Cookie。作為本網站的當前維護者，Facebook 的 Cookie 政策適用。瞭解更多資訊，包括可用控制：Cookie 政策。
