torch.cuda.reset_peak_memory_stats

torch.cuda.reset_peak_memory_stats(device=None)

重置 CUDA 記憶體分配器跟蹤的“峰值”統計資料。

參見 memory_stats() 瞭解詳情。峰值統計資料對應於每個獨立統計字典中的 “peak” 鍵。

引數

device (torch.device 或 int, 可選) – 選定的裝置。如果 device 為 None（預設），則返回當前裝置的統計資料，該裝置由 current_device() 指定。

注意

參見 記憶體管理，瞭解更多關於 GPU 記憶體管理的詳細資訊。