torch.cuda.get_device_properties

torch.cuda.get_device_properties(device=None)

取得裝置的屬性。

參數

device ( torch.device 或 int 或 str, 可選) – 要回傳其屬性的裝置。如果 device 為 None (預設值)，則使用由 current_device() 給定的當前裝置。

回傳值

裝置的屬性

回傳類型

_CudaDeviceProperties