torch.cuda.memory_usage

torch.cuda.memory_usage(device=None)

傳回過去取樣期間內，全域 (裝置) 記憶體被讀取或寫入的時間百分比，由 nvidia-smi 提供。

參數

device (torch.device 或 int, optional) – 選定的裝置。如果 device 為 None (預設值)，則傳回由 current_device() 所指定的目前裝置的統計資訊。

回傳類型

int

警告：每個樣本週期可能介於 1 秒和 1/6 秒之間，具體取決於所查詢的產品。