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
9ca12140960d6362b56b4fad09f39a1baf11c66fb14870dcd7e233a5f6293e1fCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /opt; ExposedPorts 6006/tcp, 8888/tcp
11/24/2025 10:07 PM UTC
5de5b95eaf11892e1f571e929172d03661895a8157266eef92b9d079fbf8572eARG
TARGETARCH=amd64
11/24/2025 10:07 PM UTC
2af72be034da8fb670bf516845998cfab0dff06df592382a6e8c04d0d01f9e3bARG
RC_DATE=25.11
11/24/2025 10:07 PM UTC
9d013f9dd5e1588bc68a6ee664867bfd9615f4acf9bd5573c9aba0d21774630cLABEL
com.nvidia.build.ref=9e6d0253fadd1671d152acf642783a229caa6f22
11/24/2025 10:07 PM UTC
7d66a590d5858056658203e541052e87d9144041031f889973359b8f897a1f3fARG
NVIDIA_BUILD_REF=9e6d0253fadd1671d152acf642783a229caa6f22
11/24/2025 10:07 PM UTC
6dc7c2d6447071b086c23179b0cfbf1800cc3a1ae597b287ec6e40521cc2e274LABEL
com.nvidia.build.id=237338373
11/24/2025 10:07 PM UTC
52c349790721ac5ad3378447e6fac3cca033a8ec9d5850b9ee4c84bb63be7099ENV
NVIDIA_BUILD_ID=237338373
11/24/2025 10:07 PM UTC
55b12531213fb9e2b478313e852ad6e9d1ab5956161e5fd482df6582731cc981ARG
NVIDIA_BUILD_ID=237338373
11/24/2025 10:07 PM UTC
f15ff6183d7c558d1f925dfce235ab660c975f23a4a22d06cd31e8edb42bf7beRUN
INSTALL_TE=True TE_COMMIT=release_v2.8 INSTALL_MAMBA=True MAMBA_COMMIT=6b32be06d026e170b3fdaf3ae6282c5a6ff57b06 INSTALL_BITSANDBYTES=True BITSANDBYTES_COMMIT=0.48.2 INSTALL_DEEPEP=True DEEPEP_COMMIT=ef73fd9a4c098e09929e48dfd696505ddc8e2043 BASE_IMAGE=pytorch AUTOMODEL_INSTALL=all chmod +x /opt/venv/env.sh
11/24/2025 10:07 PM UTC
00438b4bf822df517ea51f0abc9f0e166e6da6bc23e5f7a073037645c4d9c30bCOPY
<<EOF /opt/venv/env.sh
11/24/2025 10:07 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.