get_device_name¶

dragon.vm.torch.cuda.get_device_name(device=None)[source]¶

Return the name of specified device.

If device is None, the current device will be selected.

Parameters:

• device (Union[dragon.vm.torch.device, int], optional) – The device to query.

Returns:

str – The device name.