AWS Machine Learning Blog· Sunita Koppar·· 4 時間前AIスコア31
Ray Train DLCとEFAでEKS上の分散学習を行う方法
Distributed training with Ray Train DLCs and EFA on EKS
AIによる概要
Ray Train DLCは、PyTorch、CUDA、NCCL、EFAネットワークスタック、Rayを一体で検証済みのコンテナイメージで、ビルドなしで複数ノードの分散学習に使える。本サンプルでは、この公開イメージをそのまま用い、2台のg6.12xlargeインスタンス上の8基のNVIDIA L4 GPUでQwen2.5-1.5BをPyTorch FSDPによりファインチューニングする。クラスター構成はRayClusterマニフェストと変数ファイルのみで定義され、EFAはPodがリクエストすればノード間通信に利用される。
出典: AWS Machine Learning Blog · aws.amazon.com