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