torch.accelerator.is_available

torch.accelerator.is_available()

檢查當前加速器在執行時是否可用：是否已構建、所有必需的驅動程式是否可用以及至少有一個裝置是否可見。有關詳細資訊，請參閱 accelerator。

返回值

一個布林值，指示是否存在可用的 accelerator。

返回型別

bool

示例

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