oneflow.cuda.get_device_properties¶
-
oneflow.cuda.
get_device_properties
(device: Optional[Union[oneflow._oneflow_internal.device, str, int]] = None)¶ Gets the properties of a device.
The documentation is referenced from: https://pytorch.org/docs/1.10/generated/torch.cuda.get_device_properties.html.
- Parameters
device (oneflow.device or str or int) – device for which to return the properties of the device.
- Returns
the properties of the device.