Rate this Page
★★★★★

torch.cuda.get_device_name#

torch.cuda.get_device_name(device=None)[source]#

Get the name of a device.

Parameters

device (torch.device orint orstr,optional) – device for which to return thename. This function is a no-op if this argument is a negativeinteger. It uses the current device, given bycurrent_device(),ifdevice isNone (default).

Returns

the name of the device

Return type

str