torch.cuda.device_memory_used

torch.cuda.device_memory_used(device=None)

傳回 nvidia-smi 或 amd-smi 給定的已使用全域 (裝置) 記憶體，以位元組為單位。

參數

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

回傳類型

int