评价此页

torch.accelerator.device_count#

torch.accelerator.device_count()[source]#

返回当前可用的 加速器 的数量。

返回

当前可用的 加速器 的数量。

如果没有可用的加速器,则返回 0。

返回类型

int

注意

此 API 会委托给特定设备的 device_count 版本。在 CUDA 上,如果 NVML 检测成功,此 API 将不会污染 fork。否则,它会。有关更多详细信息,请参阅 多进程中的 Poison fork