torch.accelerator.is_available

torch.accelerator.is_available()

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

返回

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

返回类型

布尔值

注意

此 API 会委托给特定设备的 is_available 版本。在 CUDA 上，当环境变量 PYTORCH_NVML_BASED_CUDA_CHECK=1 设置时，此函数将不会毒化 fork。否则，它将这样做。有关更多详细信息，请参阅 多进程中的毒化 fork。

示例

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