torch.accelerator.current_accelerator

torch.accelerator.current_accelerator()

返回当前加速器的设备。

返回

返回当前加速器作为 torch.device。

返回类型

torch.device

注意

返回的 torch.device 的索引将为 None，请使用 torch.accelerator.current_device_index() 以了解当前正在使用的索引。并确保使用 torch.accelerator.is_available() 检查是否有可用的加速器。如果没有可用的加速器，此函数将引发异常。

示例

>>> if torch.accelerator.is_available():
>>>     current_device = torch.accelerator.current_accelerator()
>>> else:
>>>     current_device = torch.device("cpu")
>>> if current_device.type == 'cuda':
>>>     is_half_supported = torch.cuda.has_half
>>> elif current_device.type == 'xpu':
>>>     is_half_supported = torch.xpu.get_device_properties().has_fp16
>>> elif current_device.type == 'cpu':
>>>     is_half_supported = True