torch.cuda.get_device_properties

torch.cuda.get_device_properties(device)[源代码]

获取设备的属性。

参数

device (torch.device 或 int 或 str) – 指定返回设备属性的设备。

返回值

设备属性

返回类型

Cuda设备属性（_CudaDeviceProperties）