NVIDIA
NVIDIA
NeMo Framework Megatron Backend
Container
NVIDIA
NVIDIA
NeMo Framework Megatron Backend

NVIDIA NeMo™ framework Megatron backend supports pre-training, post-training, and reinforcement learning of LLMs and multi-modal generative AI models with state-of-the-art data processing, model training techniques, and flexible deployment options.

LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.ref=91fde638e2bd1e3c8f910c2d14c23f0142997dc8
11/04/2025 4:02 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_REF=91fde638e2bd1e3c8f910c2d14c23f0142997dc8
11/04/2025 4:02 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.id=229075953
11/04/2025 4:02 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NVIDIA_BUILD_ID=229075953
11/04/2025 4:02 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_ID=229075953
11/04/2025 4:02 AM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1RUN
NVIDIA_BIGNLP_VERSION= UV_VERSION=0.8.22 REINSTALL_NSYS=False NSYS_VERSION=2025.4.1.172 REINSTALL_CUDNN=True CUDNN_VERSION=9.13.0.50 REINSTALL_NCCL=True NCCL_VERSION=2.27.7 INSTALL_DEEPEP=True DEEPEP_COMMIT=ef73fd9a4c098e09929e48dfd696505ddc8e2043 TARGET_ARCH=x86 MCORE_REPO=https://gitlab-master.nvidia.com/ADLR/megatron-lm.git MCORE_COMMIT=04b0b576a64de4655705662d320f7f8398afd5cf MCORE_VERSION=0.14.0rc2 NEMO_COMMIT=048f57f71daae46852c066133d49234f7db85bf0 NEMO_VERSION=2.7.0rc0 NEMO_EXPORT_DEPLOY_COMMIT=6faee4f22e769e4af7e09da9e61996c2291b1a37 NEMO_EXPORT_DEPLOY_VERSION= NEMO_EVAL_COMMIT= NEMO_EVAL_VERSION= MEGATRON_BRIDGE_COMMIT=e9ee70de082891a60272221c91ecd24a8771aa3f INSTALL_RESIL=False RESIL_COMMIT=not_installed RESIL_VERSION=not_installed INSTALL_MODELOPT=True MODELOPT_VERSION=0.35.1 NEMO_RUN_COMMIT=7a13daa78e215345c2e7564a43259fb11394800a SKIP_TRTLLM=False URM_USER_ACCOUNT=donghyukc PRE_COMPILE_QUANT_KERNELS=True chmod -R a+w /workspace
11/04/2025 4:02 AM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
/workspace
11/04/2025 4:02 AM UTC
sha256:6431d4e9cc6111e8945851a3807ccf317010038428d8783f6f7e3193f3095655RUN
NVIDIA_BIGNLP_VERSION= UV_VERSION=0.8.22 REINSTALL_NSYS=False NSYS_VERSION=2025.4.1.172 REINSTALL_CUDNN=True CUDNN_VERSION=9.13.0.50 REINSTALL_NCCL=True NCCL_VERSION=2.27.7 INSTALL_DEEPEP=True DEEPEP_COMMIT=ef73fd9a4c098e09929e48dfd696505ddc8e2043 TARGET_ARCH=x86 MCORE_REPO=https://gitlab-master.nvidia.com/ADLR/megatron-lm.git MCORE_COMMIT=04b0b576a64de4655705662d320f7f8398afd5cf MCORE_VERSION=0.14.0rc2 NEMO_COMMIT=048f57f71daae46852c066133d49234f7db85bf0 NEMO_VERSION=2.7.0rc0 NEMO_EXPORT_DEPLOY_COMMIT=6faee4f22e769e4af7e09da9e61996c2291b1a37 NEMO_EXPORT_DEPLOY_VERSION= NEMO_EVAL_COMMIT= NEMO_EVAL_VERSION= MEGATRON_BRIDGE_COMMIT=e9ee70de082891a60272221c91ecd24a8771aa3f INSTALL_RESIL=False RESIL_COMMIT=not_installed RESIL_VERSION=not_installed INSTALL_MODELOPT=True MODELOPT_VERSION=0.35.1 NEMO_RUN_COMMIT=7a13daa78e215345c2e7564a43259fb11394800a SKIP_TRTLLM=False URM_USER_ACCOUNT=donghyukc PRE_COMPILE_QUANT_KERNELS=True if [ -f /usr/local/bin/torchrun ]; then sed -i '1s|#!/usr/bin/python|#!/usr/bin/env python|' /usr/local/bin/torchrun; fi
11/04/2025 4:02 AM UTC
sha256:ba904f92b3bb5ad921e626017d1533c78cc76ae945018e7915191f2312488912RUN
NVIDIA_BIGNLP_VERSION= UV_VERSION=0.8.22 REINSTALL_NSYS=False NSYS_VERSION=2025.4.1.172 REINSTALL_CUDNN=True CUDNN_VERSION=9.13.0.50 REINSTALL_NCCL=True NCCL_VERSION=2.27.7 INSTALL_DEEPEP=True DEEPEP_COMMIT=ef73fd9a4c098e09929e48dfd696505ddc8e2043 TARGET_ARCH=x86 MCORE_REPO=https://gitlab-master.nvidia.com/ADLR/megatron-lm.git MCORE_COMMIT=04b0b576a64de4655705662d320f7f8398afd5cf MCORE_VERSION=0.14.0rc2 NEMO_COMMIT=048f57f71daae46852c066133d49234f7db85bf0 NEMO_VERSION=2.7.0rc0 NEMO_EXPORT_DEPLOY_COMMIT=6faee4f22e769e4af7e09da9e61996c2291b1a37 NEMO_EXPORT_DEPLOY_VERSION= NEMO_EVAL_COMMIT= NEMO_EVAL_VERSION= MEGATRON_BRIDGE_COMMIT=e9ee70de082891a60272221c91ecd24a8771aa3f INSTALL_RESIL=False RESIL_COMMIT=not_installed RESIL_VERSION=not_installed INSTALL_MODELOPT=True MODELOPT_VERSION=0.35.1 NEMO_RUN_COMMIT=7a13daa78e215345c2e7564a43259fb11394800a SKIP_TRTLLM=False URM_USER_ACCOUNT=donghyukc PRE_COMPILE_QUANT_KERNELS=True python -m ipykernel install --name=python3 --display-name="Python 3 (ipykernel)"
11/04/2025 4:02 AM UTC
sha256:8acd6edc81ee0a2e40f48bd1f007d608c52bdc9f3f666d64947dc46f3ada3096RUN
NVIDIA_BIGNLP_VERSION= UV_VERSION=0.8.22 REINSTALL_NSYS=False NSYS_VERSION=2025.4.1.172 REINSTALL_CUDNN=True CUDNN_VERSION=9.13.0.50 REINSTALL_NCCL=True NCCL_VERSION=2.27.7 INSTALL_DEEPEP=True DEEPEP_COMMIT=ef73fd9a4c098e09929e48dfd696505ddc8e2043 TARGET_ARCH=x86 MCORE_REPO=https://gitlab-master.nvidia.com/ADLR/megatron-lm.git MCORE_COMMIT=04b0b576a64de4655705662d320f7f8398afd5cf MCORE_VERSION=0.14.0rc2 NEMO_COMMIT=048f57f71daae46852c066133d49234f7db85bf0 NEMO_VERSION=2.7.0rc0 NEMO_EXPORT_DEPLOY_COMMIT=6faee4f22e769e4af7e09da9e61996c2291b1a37 NEMO_EXPORT_DEPLOY_VERSION= NEMO_EVAL_COMMIT= NEMO_EVAL_VERSION= MEGATRON_BRIDGE_COMMIT=e9ee70de082891a60272221c91ecd24a8771aa3f INSTALL_RESIL=False RESIL_COMMIT=not_installed RESIL_VERSION=not_installed INSTALL_MODELOPT=True MODELOPT_VERSION=0.35.1 NEMO_RUN_COMMIT=7a13daa78e215345c2e7564a43259fb11394800a SKIP_TRTLLM=False URM_USER_ACCOUNT=donghyukc PRE_COMPILE_QUANT_KERNELS=True if [ $PRE_COMPILE_QUANT_KERNELS = "True" ]; then python -c "import modelopt.torch.quantization.extensions as ext; ext.precompile()"; fi
11/04/2025 4:02 AM 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.