torch.accelerator.is_available

torch.accelerator.is_available()

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

返回值

一个布尔值，指示是否存在可用的 accelerator。

返回类型

bool

示例

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