ray.train.TrainContext.get_node_rank

abstract TrainContext.get_node_rank() → int

Get the rank of this node.

Example

import ray.train
from ray.train.torch import TorchTrainer

def train_fn_per_worker():
    print(ray.train.get_context().get_node_rank())

trainer = TorchTrainer(
    train_fn_per_worker,
    scaling_config=ray.train.ScalingConfig(num_workers=1),
)
trainer.fit()