torch.accelerator.device_count

torch.accelerator.device_count()

返回当前可用的 accelerator 的数量。

返回:

当前可用的 accelerator 的数量。

如果没有可用的 accelerator，则返回 0。

返回类型:

int

注意

此 API 将委托给特定设备的 device_count 版本。在 CUDA 上，如果 NVML 发现成功，此 API 将不会毒化 fork。否则，它会。有关更多详细信息，请参阅 多进程中的毒化 fork。