ray.util.accelerators.tpu.get_current_pod_worker_count

ray.util.accelerators.tpu.get_current_pod_worker_count() → int | None

计算与该工作者所属的 TPU pod 相关联的工作者数量。

返回:

TPU pod 中的总工作线程数。如果工作线程不属于 TPU pod，则返回 None。

PublicAPI (alpha): 此API处于alpha阶段，可能在稳定之前发生变化。