Installation

LightlyTrain is available on PyPI and can be installed via pip or other package managers.

Warning

To successfully install LightlyTrain the python version has to be >= 3.8 and <= 3.12 .

pip install lightly-train

To update to the latest version, run:

pip install --upgrade lightly-train

See Docker for Docker installation instructions.

Optional Dependencies

LightlyTrain has optional dependencies that are not installed by default. The following dependencies are available:

Logging

• tensorboard: For logging to TensorBoard

• wandb: For logging to Weights & Biases

Model Support

• super-gradients: For SuperGradients models

• timm: For TIMM models

• ultralytics: For Ultralytics models

To install optional dependencies, run:

pip install lightly-train[tensorboard]

Or for multiple optional dependencies:

pip install lightly-train[tensorboard,timm]