这是Kubeflow训练框架v1版本下的PyTorch与DeepSpeed结合的演示容器镜像,包含使用PyTorch和DeepSpeed进行分布式训练的示例代码及相关配置,可用于快速体验基于该组合的机器学习训练流程。
ghcr.io/kubeflow/training-v1/pytorch-deepspeed-demo:latest