oneflow.cuda.get_device_name

oneflow.cuda.get_device_name(device: Optional[Union[oneflow._oneflow_internal.device, str, int]] = None)str

Gets the name of a device.

The documentation is referenced from: https://pytorch.org/docs/1.10/generated/torch.cuda.get_device_name.html.

Parameters

device (oneflow.device or int or str, optional) – device for which to return the name. It uses the current device, given by current_device(), if device is None (default).

Returns

the name of the device

Return type

str