torch.xpu.max_memory_allocated¶
- torch.xpu.max_memory_allocated(device=None)[原始碼][原始碼]¶
返回給定裝置上張量佔用的最大 GPU 記憶體量（位元組）。
預設情況下，此函式返回自程式開始以來分配的峰值記憶體。可以使用
reset_peak_memory_stats()來重置跟蹤此指標的起始點。例如，這兩個函式可以測量訓練迴圈中每次迭代的峰值分配記憶體使用量。- 引數
device (torch.device 或 int 或 str，可選) – 選擇的裝置。如果
device是None（預設值），則返回current_device()給出的當前裝置的統計資訊。- 返回型別
