ludwigai/ludwig-ray-gpu repository overview

⁠ludwigai/ludwig-ray-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 plus Ray for distributed training: text, image, and audio features, visualizations, hyperparameter optimization, model serving, and distributed training via Ray 2.54.

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-ray-gpu experiment \
  --config /data/config.yaml --dataset /data/train.csv --backend ray

⁠Links

• GitHub⁠
• Documentation⁠
• CPU variant⁠
• Non-Ray GPU variant⁠