Commit aa121a0
Bhanu Teja Goshikonda
fix(pytorch): pass --disable-ngc to EFA installer for 1.48+
Starting with EFA 1.48, the installer auto-detects NGC containers via
/opt/nvidia/nvidia_entrypoint.sh (present in nvidia/cuda:*-amzn2023 base
images) and skips installing the AL2023 libnccl-ofi RPM. This left the
pytorch runtime image without libnccl-net.so, breaking every unit /
sanity / single-gpu / efa test that asserts the OFI plugin is present.
The sibling scripts/docker/common/install_efa_amzn2023.sh already has
this fix (lines 37-48); mirror the version-gated --disable-ngc flag here
so aws-ofi-nccl installs regardless of the base-image NGC detection.
Only affects pytorch - sglang, vllm, and vllm-omni use the common
installer via their own paths.1 parent 66b9fe3 commit aa121a0
1 file changed
Lines changed: 7 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
10 | 10 | | |
11 | 11 | | |
12 | 12 | | |
13 | | - | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
14 | 20 | | |
15 | 21 | | |
16 | 22 | | |
| |||
0 commit comments