Installation

Platform Compatibility

LightlyTrain is officially supported on

• Linux: CPU or CUDA

• MacOS: CPU only

We are planning to support MPS for MacOS, and to support Windows in the future.

Installation from PyPI

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

• wandb: For logging to Weights & Biases

Model Support

• rfdetr: For RF-DETR models

• super-gradients: For SuperGradients models

• timm: For TIMM models

• ultralytics: For Ultralytics models

To install optional dependencies, run:

pip install "lightly-train[wandb]"

Or for multiple optional dependencies:

pip install "lightly-train[wandb,timm]"

Hardware Recommendations

An example hardware setup and its performance when using LightlyTrain is provided in Hardware Recommendations.