bitnamilegacy/deepspeed 提供 Microsoft DeepSpeed 分布式训练环境，ZeRO 优化器分片大模型参数，多 GPU 并行训练。定位为 GPU 分布式训练环境，需 --gpus all 与 NCCL；docker GPU 集群训练，非 CPU-only K8s 常规场景。

快速启动

docker run --rm \
  --gpus all \
  docker.io/bitnamilegacy/deepspeed:latest \
  --help

推荐实践

docker run --rm \
  --gpus all \
  -v /data/models:/models \
  docker.io/bitnamilegacy/deepspeed:0.14.0 \
  deepspeed --num_gpus 4 train.py

核心参数说明

• --gpus all — 多 GPU 训练
• deepspeed --num_gpus — 分布式训练启动
• 优化 — ZeRO-1/2/3 参数分片
• 定位 — GPU 训练 batch 环境