跳到正文
AWS Machine Learning Blog· Sunita Koppar·· 12 天前AI 评分42

AWS 发布 Ray Train DLC 以在 EKS 上利用 EFA 实现分布式训练

Distributed training with Ray Train DLCs and EFA on EKS

AI 导读

AWS 推出 Ray Train 深度学习容器(DLC),预集成 PyTorch 2.13.0、CUDA 13.0.2 及 EFA 网络栈,消除分布式训练环境配置障碍。该镜像通过 KubeRay 在 EKS 上部署,支持跨节点 EFA 通信,示例展示了在 8 块 NVIDIA L4 GPU 上使用 FSDP 微调 Qwen2.5-1.5B 模型。

来源:AWS Machine Learning Blog · aws.amazon.com