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."