AWS Machine Learning Blog· Sunita Koppar·· 4 小时前AI 评分52
使用 Ray Train DLC 和 EFA 在 EKS 上进行分布式训练
Distributed training with Ray Train DLCs and EFA on EKS
AI 导读
AWS 演示如何使用 Ray Train DLC、KubeRay 和 EFA 在 Amazon EKS 上运行分布式 PyTorch 训练,而无需自行构建容器镜像。示例通过两个 g6.12xlarge 实例和 8 张 NVIDIA L4 GPU,以 FSDP 微调 Qwen2.5-1.5B,并提供训练代码、RayCluster 配置、部署脚本及环境要求。
来源:AWS Machine Learning Blog · aws.amazon.com