ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
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.

NVIDIA AI Enterprise Supported
LayerLabelCreated
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
/opt
04/16/2025 9:22 PM UTC
sha256:1e75432dba30531e4dac86600bfc82c072c1800c1e6d755d8738643d57ab9ed7RUN
NVIDIA_BIGNLP_VERSION= TRT_VER=10.9.0.34 CUDA_VER=12.8 CUDNN_VER=9.8.0.87-1 NCCL_VER=2.25.1-1+cuda12.8 CUBLAS_VER=12.8.4.1-1 NVRTC_VER=12.8.93-1 TRT_LLM_COMMIT=fe85b725df3e769b87a8fdb63154276f87032080 TRT_LLM_VERSION=0.17.1 REINSTALL_NSYS=False NSYS_VERSION=2025.1.1.65 REINSTALL_APEX=False APEX_COMMIT=e13873debc4699d39c6861074b9a3b2a02327f92 MCORE_COMMIT=baf1d05e3c518f71b0d41ae42381ecbada0469ab MCORE_VERSION=0.12.0rc0 SRC_DIR=/src/tensorrt_llm CPP_BUILD_DIR=/src/tensorrt_llm/cpp/build TARGET_ARCH=x86 INSTALL_VLLM=True ALIGNER_COMMIT=0d72d2829d34c07cd8685348c0f37f6b6b7e5834 ALIGNER_VERSION=0.7.0 CURATOR_COMMIT=r0.8.0 INSTALL_CURATOR=True CURATOR_VERSION=0.8.0rc4.dev0 NEMO_COMMIT=101ffd44487c4a230cb5b9022ec97823ddb8f610 NEMO_VERSION=2.4.0rc0 INSTALL_RESIL=False RESIL_COMMIT=not_installed RESIL_VERSION=not_installed INSTALL_MODELOPT=False MODELOPT_VERSION= NEMO_RUN_COMMIT=r0.4.0 PRE_COMPILE_QUANT_KERNELS=True NEMO_FW_PIPELINE_ID=27072633 NVIDIA_BUILD_ID=148941828 NVIDIA_BUILD_REF= pip uninstall -y triton &&
  pip install triton==3.2
04/16/2025 9:22 PM UTC
sha256:7ec600a0a2fefd8ef043dfde64958e6a9fb2906584a5622c1be873b9201c1c6eRUN
NVIDIA_BIGNLP_VERSION= TRT_VER=10.9.0.34 CUDA_VER=12.8 CUDNN_VER=9.8.0.87-1 NCCL_VER=2.25.1-1+cuda12.8 CUBLAS_VER=12.8.4.1-1 NVRTC_VER=12.8.93-1 TRT_LLM_COMMIT=fe85b725df3e769b87a8fdb63154276f87032080 TRT_LLM_VERSION=0.17.1 REINSTALL_NSYS=False NSYS_VERSION=2025.1.1.65 REINSTALL_APEX=False APEX_COMMIT=e13873debc4699d39c6861074b9a3b2a02327f92 MCORE_COMMIT=baf1d05e3c518f71b0d41ae42381ecbada0469ab MCORE_VERSION=0.12.0rc0 SRC_DIR=/src/tensorrt_llm CPP_BUILD_DIR=/src/tensorrt_llm/cpp/build TARGET_ARCH=x86 INSTALL_VLLM=True ALIGNER_COMMIT=0d72d2829d34c07cd8685348c0f37f6b6b7e5834 ALIGNER_VERSION=0.7.0 CURATOR_COMMIT=r0.8.0 INSTALL_CURATOR=True CURATOR_VERSION=0.8.0rc4.dev0 NEMO_COMMIT=101ffd44487c4a230cb5b9022ec97823ddb8f610 NEMO_VERSION=2.4.0rc0 INSTALL_RESIL=False RESIL_COMMIT=not_installed RESIL_VERSION=not_installed INSTALL_MODELOPT=False MODELOPT_VERSION= NEMO_RUN_COMMIT=r0.4.0 PRE_COMPILE_QUANT_KERNELS=True NEMO_FW_PIPELINE_ID=27072633 NVIDIA_BUILD_ID=148941828 NVIDIA_BUILD_REF= rm -rf mamba &&
  git clone https://github.com/state-spaces/mamba.git &&
  cd mamba &&
  MAMBA_FORCE_BUILD=TRUE pip install . --no-build-isolation
04/16/2025 9:22 PM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
/opt
04/16/2025 9:19 PM UTC
sha256:d04a628cf44157765962fe1cd41b53af7c77c9f7fbd916e430f8a25b9fbc1083RUN
NVIDIA_BIGNLP_VERSION= TRT_VER=10.9.0.34 CUDA_VER=12.8 CUDNN_VER=9.8.0.87-1 NCCL_VER=2.25.1-1+cuda12.8 CUBLAS_VER=12.8.4.1-1 NVRTC_VER=12.8.93-1 TRT_LLM_COMMIT=fe85b725df3e769b87a8fdb63154276f87032080 TRT_LLM_VERSION=0.17.1 REINSTALL_NSYS=False NSYS_VERSION=2025.1.1.65 REINSTALL_APEX=False APEX_COMMIT=e13873debc4699d39c6861074b9a3b2a02327f92 MCORE_COMMIT=baf1d05e3c518f71b0d41ae42381ecbada0469ab MCORE_VERSION=0.12.0rc0 SRC_DIR=/src/tensorrt_llm CPP_BUILD_DIR=/src/tensorrt_llm/cpp/build TARGET_ARCH=x86 INSTALL_VLLM=True ALIGNER_COMMIT=0d72d2829d34c07cd8685348c0f37f6b6b7e5834 ALIGNER_VERSION=0.7.0 CURATOR_COMMIT=r0.8.0 INSTALL_CURATOR=True CURATOR_VERSION=0.8.0rc4.dev0 NEMO_COMMIT=101ffd44487c4a230cb5b9022ec97823ddb8f610 NEMO_VERSION=2.4.0rc0 INSTALL_RESIL=False RESIL_COMMIT=not_installed RESIL_VERSION=not_installed INSTALL_MODELOPT=False MODELOPT_VERSION= NEMO_RUN_COMMIT=r0.4.0 PRE_COMPILE_QUANT_KERNELS=True NEMO_FW_PIPELINE_ID=27072633 NVIDIA_BUILD_ID=148941828 NVIDIA_BUILD_REF= pip install --upgrade wandb setuptools
04/16/2025 9:19 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.ref=
04/16/2025 9:19 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_REF
04/16/2025 9:19 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.id=148941828
04/16/2025 9:19 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NVIDIA_BUILD_ID=148941828
04/16/2025 9:19 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_ID
04/16/2025 9:19 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.