Skip to content

scope ssh keypair to pytorch training, move non-base images to efa latest, patch cves - #6785

Open
jinyan-li1 wants to merge 27 commits into
mainfrom
scope-ssh-keypair-pytorch
Open

jinyan-li1 wants to merge 27 commits into
mainfrom
scope-ssh-keypair-pytorch

Conversation

@jinyan-li1

@jinyan-li1 jinyan-li1 commented Sep 23, 2026 •

Copy link
Copy Markdown
Contributor

Bundles three changes to the EFA-enabled images:

1. Scope the SSH keypair to pytorch training
Generate the SSH keypair only in pytorch-cuda (via configure_ssh) instead of the shared EFA install scripts, so it's no longer created in every EFA-based image. pytorch-cuda keeps it (uses mpirun-over-SSH for multi-node launch); every other EFA image drops it — including ray-train, which coordinates multi-node via Ray rather than SSH.

2. Opt non-base images into EFA latest
install_efa.sh / install_efa_amzn2023.sh now accept efa_version: "latest" alongside a pinned number, and log the resolved version from /opt/amazon/efa_installed_packages. All non-base EFA images move to latest; base stays pinned.

3. CVE remediation
Cache-busted dnf upgrade --security on the base and a few downstream images, a couple of pure-Python package bumps, and ECR scan allowlist updates.

@jinyan-li1 jinyan-li1 changed the title scope ssh keypair to pytorch training image scope ssh keypair to pytorch training, move non-base images to efa latest, patch cves Oct 1, 2026

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant