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
849220b9213e6d9e9e08e0812e8f82e45dcb92b791669ab004f13f8d0bf7910aCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
03/31/2025 4:46 PM UTC
948b76349a9cb746298fc1f18de95da3371a813c13d673d7a46ad1c6bea122c4RUN
rm -rf /opt/NeMo
03/05/2025 8:10 AM UTC
1aa785d1f3a3bbeb002f9fad5f995cb982e16e76cfb8c1f4e25e542677f27ae6RUN
git clone --single-branch --branch cosmos1.0.1 https://github.com/NVIDIA/Megatron-LM.git /opt/megatron-lm &&
  cd /opt/megatron-lm &&
  git checkout 86dad2c39a52c241e372b774f07041201f8a57f4 &&
  pip install -e . &&
  cd megatron/core/datasets &&
  make
03/05/2025 8:10 AM UTC
c1be2d00d8e5b39560737bfd28c7cadc397f58c1e71879066f1f38ecf07a7c69RUN
rm -rf /opt/megatron-lm
03/05/2025 8:10 AM UTC
2a5e64123b3e1c5c7132e54a0769c0d43602d47f97060919eb996836b4264c55RUN
pip install moviepy==1.0.3
03/05/2025 8:10 AM UTC
9936678482fdedde45b6c542c56a3b39355d1a813950f83d64fdb369b4d8fc88RUN
pip install --no-cache-dir imageio[ffmpeg] pyav iopath better_profanity peft lru-dict pyquaternion git+https://github.com/NVlabs/Pytorch_Retinaface.git@b843f45 git+https://github.com/NVIDIA/NeMo-Run.git@f07877f80dbfc91c4d37683864cf2552b96d62f1
03/05/2025 8:10 AM UTC
db3acdbacfc9ba0f4bf8d8d5248eebb05ddc7968afe9ed444ee09344cfbf8ebbLABEL
com.nvidia.build.ref=
02/18/2025 9:19 PM UTC
3d4f8fd7ae3271ede5731896ae0b216ca697f46221d96881c872adb836b805e5ARG
NVIDIA_BUILD_REF
02/18/2025 9:19 PM UTC
ca887158ae213227f59ac1877f074ce32814f9865b175a0fe8d0c8b9e63218faLABEL
com.nvidia.build.id=134983853
02/18/2025 9:19 PM UTC
64250a97d846e3847eb6af3ce0c8c65fcbad7e50afe4e1ad26e4f373183cbe60ENV
NVIDIA_BUILD_ID=134983853
02/18/2025 9:19 PM UTC
...