评价此页

torch.accelerator.is_available#

torch.accelerator.is_available()[源代码]#

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

返回

一个布尔值,指示是否有可用的加速器

返回类型

布尔值

注意

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

示例

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