AWS Machine Learning Blog· Sunita Koppar·· 3 小时前精选AI 评分52
在 EKS 上用 Ray Train DLC 与 EFA 做分布式训练
Distributed training with Ray Train DLCs and EFA on EKS
AI 导读
AWS 发布教程,演示直接用已发布的 Ray Train DLC 镜像(train-ml-cuda-v1.1)在 EKS 上通过 KubeRay 启动集群,用 PyTorch FSDP 在两台 g6.12xlarge 的 8 张 NVIDIA L4 上微调 Qwen2.5-1.5B,全程不构建镜像。
推荐理由
AWS 官方给出不构建镜像即可在 EKS 上跑多机 FSDP 训练的完整路径,含清单、脚本与 EFA 传输配置。
来源:AWS Machine Learning Blog · aws.amazon.com