torch.cuda.power_draw

torch.cuda.power_draw(device=None)

傳回 GPU 感測器在過去的取樣期間內的平均功耗，單位為 mW（毫瓦）。

取樣期間由 nvidia-smi 針對 Fermi 或更新且完全支援的裝置給定。

參數

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

回傳類型

int

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