NVIDIA
NVIDIA
nemo-automodel
Container
NVIDIA
NVIDIA
nemo-automodel

NVIDIA NeMo™ AutoModel accelerates LLM and VLM training and fine‑tuning with PyTorch DTensor‑native SPMD, day‑0 Hugging Face support, and optimized parallelism from single‑ to multi‑node scale.

LayerLabelCreated
5b51e728c340d9e9ea335b6c200d0f9710878c86d905cbfa177840f0822de05fCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /opt/Automodel; ExposedPorts 6006/tcp, 8888/tcp
06/29/2026 7:12 PM UTC
09a627092722025e1bfcd46151abfa289056b289be32801319ed4eef22963b9cARG
TARGETARCH=amd64
06/29/2026 7:12 PM UTC
1d1304fcbdacb38e6b916a1894f8bd619a714be957f9246685978d0fae8e844dARG
RC_DATE=26.06
06/29/2026 7:12 PM UTC
ff47aac8fa2358cc29f8f86be17c082f0fd33bfc3a6c97f18c4c1a50fa6c77a9LABEL
com.nvidia.build.ref=79a86c9ca724782a95167d0f875625c97b6178c6
06/29/2026 7:12 PM UTC
908104b71034a70cd90968c6f000915fc6091bb2451974dd7cd071c7191dd8a5ARG
NVIDIA_BUILD_REF=79a86c9ca724782a95167d0f875625c97b6178c6
06/29/2026 7:12 PM UTC
6ea220e6a186f445499931d14c0ad9fff90bfa9a054fc3f8b9e38e28f692d6dbLABEL
com.nvidia.build.id=350786066
06/29/2026 7:12 PM UTC
e30e433f322ee59fb9779ccd1547c3c9f4e10f359998108ff44117125fdda91eENV
NVIDIA_BUILD_ID=350786066
06/29/2026 7:12 PM UTC
e3c769583b1f43086b0fbb58f99ed13f411f566909f60867f07a2ee4870cc8e6ARG
NVIDIA_BUILD_ID=350786066
06/29/2026 7:12 PM UTC
0ca8bcd65604f120a4ee56215a37e5a23ffcbda7e0b97b48cacab4c9a5af9b53RUN
INSTALL_TE=True TE_COMMIT=release_v2.14 DEEPEP_COMMIT=42144303752422ade37f24bca9e2dde12df70e09 INSTALL_BITSANDBYTES=True BITSANDBYTES_COMMIT=0.49.2 INSTALL_UCCL_EP=False BASE_IMAGE=pytorch AUTOMODEL_INSTALL=all UV_SYNC_ARGS=--locked chmod +x /opt/venv/env.sh
06/29/2026 7:12 PM UTC
93f860037744ec97000012aad94e3b0db7a0ad2331740e22b51ae7b2f382b663COPY
<<EOF /opt/venv/env.sh
06/29/2026 7:12 PM UTC
...

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.