ray.train.lightning.RayLightningEnvironment#

class ray.train.lightning.RayLightningEnvironment(*args: Any, **kwargs: Any)[source]#

Bases: pytorch_lightning.plugins.environments.LightningEnvironment

Setup Lightning DDP training environment for Ray cluster.

PublicAPI (beta): This API is in beta and may change before becoming stable.

Methods