AWS Machine Learning Blog· Sunita Koppar·· 4 小时前AI 评分38
在 EKS 上使用 Ray Train DLC 与 EFA 进行分布式训练
Distributed training with Ray Train DLCs and EFA on EKS
AI 导读
AWS 构建的 Ray Train DLC 是预置 EFA 网络栈的 PyTorch 分布式训练容器镜像,无需自建镜像即可在 EKS 上运行多节点训练。示例用 PyTorch FSDP 在两个 g6.12xlarge 实例的 8 张 NVIDIA L4 GPU 上微调 Qwen2.5-1.5B。
来源:AWS Machine Learning Blog · aws.amazon.com