torch.cuda.get_device_properties

torch.cuda.get_device_properties(device=None)

獲取裝置的屬性。

引數

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

返回值

裝置的屬性

返回型別

_CudaDeviceProperties