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
2863f10670303e45eca5b637304184b8cee20eef182c15bfb749aa86edaee1d8CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /opt; ExposedPorts 6006/tcp, 8888/tcp
08/14/2025 9:35 PM UTC
41bffa984a811ddfb6255e8b8ad92669d9bc14da4044b26ca144888b88ec651aENV
PYTHONPATH=/opt/Export-Deploy:/opt/venv/lib/python3.12/site-packages/::/opt/megatron-lm:/opt/NeMo
08/14/2025 9:35 PM UTC
aa9088f0f911674ebb06fda011cf094de0fd192205e5e77bbb2f3d79c69966afRUN
NEMO_COMMIT=2e19aebd8c8fa9ff7ce9b5076ce130404713443c rm -rf NeMo &&
  git clone https://github.com/NVIDIA/NeMo.git &&
  cd NeMo &&
  git fetch --all &&
  git checkout $NEMO_COMMIT &&
  echo "Container built with NeMo commit hash: $(git rev-parse HEAD)" &&
  pip uninstall -y nemo_toolkit sacrebleu triton &&
  sed -i "/mamba-ssm/d" requirements/requirements_nlp.txt &&
  cd nemo/collections/nlp/data/language_modeling/megatron &&
  make
08/14/2025 9:35 PM UTC
7a1883e23b9c197a17fc72c8f42d98ee96f3498cea6ca22d616d6550d356b64bWORKDIR
/opt
08/12/2025 7:50 PM UTC
fff58d5097c9a36afbffa549eb3a95c46af792b0cb9c03e516536ab706f30508ARG
NEMO_COMMIT=2e19aebd8c8fa9ff7ce9b5076ce130404713443c
08/12/2025 7:50 PM UTC
6c23a0124c937cc11f1c3bce20874aea1db3938e330114e35a1eaa610d571243LABEL
com.nvidia.build.ref=250aafd22bf377cbc368a206994bce682f1f1369
08/07/2025 1:53 PM UTC
6a7a7a2135796417377b5134463708567f791113a045f759b4503583515a075cARG
NVIDIA_BUILD_REF=250aafd22bf377cbc368a206994bce682f1f1369
08/07/2025 1:53 PM UTC
9d1bcfcd3fadd4cc04a345f76d397afeaad1665d03bddbcaac87a50303cb049fLABEL
com.nvidia.build.id=196948111
08/07/2025 1:53 PM UTC
a70a5225d416422d5f05284738eed304d17f1b9ab6270dde89db313e56192695ENV
NVIDIA_BUILD_ID=196948111
08/07/2025 1:53 PM UTC
dca4f9434b08bca8544810ae77d844de137f8e138500e981fc480b5714b7e52eARG
NVIDIA_BUILD_ID=196948111
08/07/2025 1:53 PM UTC
...