torch.xpu.memory_allocated¶
- torch.xpu.memory_allocated(device=None)[source][source]¶
返回給定裝置上張量當前佔用的裝置記憶體量（以位元組為單位）。
- 引數
device (torch.device 或 int 或 str，可選) – 所選裝置。如果
device為None(預設值)，則返回由current_device()給出的當前裝置的統計資訊。- 返回型別
注意
這可能小於 xpu-smi 中顯示的記憶體量，因為一些未使用的記憶體可能由快取分配器持有，並且需要在 GPU 上建立一些上下文。
