torch.cuda.get_device_name¶
- torch.cuda.get_device_name(device=None)[source][source]¶
Get the name of a device.
- Parameters
device (torch.device or int or str, optional) – device for which to return the name. This function is a no-op if this argument is a negative integer. It uses the current device, given by
current_device(), ifdeviceisNone(default).- Returns
the name of the device
- Return type
