ray.util.accelerators.tpu.get_current_pod_name

ray.util.accelerators.tpu.get_current_pod_name() → str | None

返回该工作节点所属的 TPU pod 的名称。

返回:

TPU pod 的名称。如果不是 TPU pod 的一部分，则返回 None。

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