torch.accelerator.device_count

torch.accelerator.device_count()

返回當前可用的 加速器 的數量。

返回

當前可用的 加速器 的數量。

如果沒有可用的加速器，則返回 0。

返回型別

int

注意

此 API 會委託給特定裝置的 device_count 版本。在 CUDA 上，如果 NVML 發現成功，此 API 不會汙染 fork。否則，它會。有關更多詳細資訊，請參閱 多程序中的汙染 fork。