torch.xpu.max_memory_reserved¶
- torch.xpu.max_memory_reserved(device=None)[原始碼][原始碼]¶
傳回特定裝置的快取分配器所管理的最大 GPU 記憶體 (以位元組為單位)。
預設情況下，這會傳回自程式開始以來快取記憶體的最高峰值。可以使用
reset_peak_memory_stats()來重置追蹤此指標的起點。例如，這兩個函數可以測量訓練迴圈中每次迭代的快取記憶體峰值。- 參數
device (torch.device 或 int 或 str, optional) – 選擇的裝置。如果
device為None(預設)，則傳回由current_device()給定的目前裝置的統計資訊。- 回傳類型
