XLAEnvironment

class lightning.pytorch.plugins.environments.XLAEnvironment(*args, **kwargs)

기반 클래스: lightning.fabric.plugins.environments.cluster_environment.ClusterEnvironment

Cluster environment for training on a TPU Pod with the PyTorch/XLA library.

A list of environment variables set by XLA can be found here.

static detect()

Detects the environment settings corresponding to this cluster and returns True if they match.

반환 형식

bool

global_rank()

The rank (index) of the currently running process across all nodes and devices.

반환 형식

int

local_rank()

The rank (index) of the currently running process inside of the current node.

반환 형식

int

node_rank()

The rank (index) of the node on which the current process runs.

반환 형식

int

world_size()

The number of processes across all devices and nodes.

반환 형식

int

property creates_processes_externally: bool

Whether the environment creates the subprocesses or not.

반환 형식

bool

property main_address: str

The main address through which all processes connect and communicate.

반환 형식

str

property main_port: int

An open and configured port in the main node through which all processes communicate.

반환 형식

int