Skip to content
AWS Machine Learning Blog· Sunita Koppar·· 4 h ago

Distributed training with Ray Train DLCs and EFA on EKS

Distributed training with Ray Train DLCs and EFA on EKS

AI summary

The Ray Train Deep Learning Container ships the full CUDA-through-EFA stack, so you build no image. This post walks through a reproducible sample that fine-tunes Qwen2.5-1.5B with PyTorch FSDP across eight GPUs on two EFA-enabled g6.12xlarge instances on Amazon EKS, using the container exactly as published.

Selection record

Not admittedSum of both 69 < twice the threshold 120

Source tier
Official, first-hand; this tier's threshold is 60
Pre-filter
passed:分布式训练Ray Train与FSDP实操

A model scores each item twice, independently, against one written standard, out of 100. An item is admitted only when the two scores add up to twice the threshold. The threshold is set per source tier.

Source: AWS Machine Learning Blog · aws.amazon.com