ludwigai/ludwig-gpu repository overview

⁠ludwigai/ludwig-gpu

Ludwig is a declarative deep learning framework that allows you to train, evaluate, and deploy models without writing boilerplate code.

This image includes the full set of Ludwig dependencies: text, image, and audio features, visualizations, hyperparameter optimization, and model serving.

PyTorch: GPU-enabled (CUDA 12.4)

⁠Tags

• latest — latest stable release
• X.Y.Z — exact release version (e.g. 0.13.0)
• X.Y — latest patch for a minor version (e.g. 0.13)

⁠Usage

docker run --gpus all -v /path/to/data:/data ludwigai/ludwig-gpu experiment \
  --config /data/config.yaml --dataset /data/train.csv

⁠Links

• GitHub⁠
• Documentation⁠
• CPU variant⁠
• Ray GPU distributed variant⁠