torch.cuda.get_device_name

torch.cuda.get_device_name(device=None)

获取设备的名称。

Parameters

设备 (torch.device 或 整数, 可选) – 返回名称的设备。如果此参数为负整数，则此函数不执行任何操作。如果device为None（默认），则使用当前设备，由current_device()给出。

Returns

设备的名称

Return type

str