Environments

ClusterEnvironment — Specification of a cluster environment.
KubeflowEnvironment — Environment for distributed training using the PyTorchJob operator from Kubeflow.
LightningEnvironment — The default environment used by Lightning for a single node or free cluster (not managed).
LSFEnvironment — An environment for running on clusters managed by the LSF resource manager.
MPIEnvironment — An environment for running on clusters with processes created through MPI.
SLURMEnvironment — Cluster environment for training on a cluster managed by SLURM.
TorchElasticEnvironment — Environment for fault-tolerant and elastic training with torchelastic
XLAEnvironment — Cluster environment for training on a TPU Pod with the PyTorch/XLA library.

Was this page helpful?