Shortcuts

torch.cuda.get_device_properties

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

获取设备的属性。

Parameters

设备 (torch.device整数字符串) – 要返回其属性的设备。

Returns

设备的属性

Return type

_CudaDeviceProperties