torch.accelerator.is_available

torch.accelerator.is_available()[source][source]

检查当前加速器在运行时是否可用：它已构建，所有必要的驱动程序都可用，并且至少有一个设备可见。有关详细信息，请参阅加速器。

返回值:

表示是否存在可用的加速器。

返回类型:

布尔型

示例：

>>> assert torch.accelerator.is_available() "No available accelerators detected."