本文へスキップ
原文
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