torch.cuda.utilization

torch.cuda.utilization(device=None)

傳回過去取樣期間內，一個或多個核心在 GPU 上執行的時間百分比，此資料由 nvidia-smi 提供。

參數

device (torch.device 或 int, optional) – 選擇的裝置。如果 device 為 None (預設)，則傳回目前裝置的統計資料，由 current_device() 提供。

回傳類型

int

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