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
fedf50398d86803d72014c2d7b2a4c702a7797007f3869ac823cc6da1f6c6151CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
01/10/2024 7:04 PM UTC
a904c4dc28f98a2c63cb724234ed3858aadbdfc5cd52bd2d11d54f9696149336RUN
NEMO_VERSION=1.22.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
01/10/2024 7:04 PM UTC
31342712274e6e8958323d63dfa3c30c555f1e5f28c89601fedaef1f1e01471aCOPY
tutorials /workspace/nemo/tutorials
01/10/2024 7:04 PM UTC
2359d1cd75aa9ece1375356c745a164e3d98f8eb3d8943a9548a786f94c69a47COPY
tests /workspace/nemo/tests
01/10/2024 7:04 PM UTC
590f2503c723ad67e7e9b48d9f49d2ad2c94b2555825dc07a02880257dc71078COPY
examples /workspace/nemo/examples
01/10/2024 7:04 PM UTC
cdae06e2952e7900fac87c9b4c52cd47015f9565dad392f07d019fc604f74741COPY
scripts /workspace/nemo/scripts
01/10/2024 7:04 PM UTC
8ebfeca7dab222073f3c1704c8ad1c7209909d99143068bf4b22fae61b01bef9WORKDIR
/workspace/nemo
01/10/2024 7:04 PM UTC
6013005d8befdf6565457f0fa5389382e360b4306b57229b5029e7ea39125ed0RUN
NEMO_VERSION=1.22.0 python -c "import nemo.collections.nlp as nemo_nlp" &&
  python -c "import nemo.collections.tts as nemo_tts" &&
  python -c "import nemo_text_processing.text_normalization as text_normalization"
01/10/2024 7:04 PM UTC
508c680f31a80d2390d2bc498e80803542372face96aa40a59d3293a9d78affeRUN
NEMO_VERSION=1.22.0 cd /tmp/nemo &&
  pip install ".[all]"
01/10/2024 7:04 PM UTC
9d7ef7334220996c428711f6a4ddeb06f75d0e2538908f08efd6d70960045421RUN
NEMO_VERSION=1.22.0 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
01/10/2024 7:03 PM UTC
...