torch.cuda.get_device_properties

torch.cuda.get_device_properties(device=None)

獲取裝置的屬性。

引數

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

返回

裝置的屬性

返回型別

_CudaDeviceProperties