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
477d43b5451b456bf135290b732cc10130702b15dbc27db0cab8b19910e1ec1bCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
02/27/2024 5:37 AM UTC
46bf60eea38ef77bc48016726f12a43668b5bc1c3cfcafdb5b36588b5f7e90a4RUN
NEMO_VERSION=1.23.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
02/27/2024 5:37 AM UTC
49bad04dc9c3da473110d04d45dab5a748b92cd7c7c9d979eac7a1ffe3b068c8COPY
tutorials /workspace/nemo/tutorials
02/27/2024 5:37 AM UTC
f37e056d169929b9b7e3846f70a6f7bdb15735d59ac8bd8473e8ede902337902COPY
tests /workspace/nemo/tests
02/27/2024 5:37 AM UTC
45fba5220553221801906c128cf7356bcf00721282fcfc20017c1201a0b6f27fCOPY
examples /workspace/nemo/examples
02/27/2024 5:37 AM UTC
bfd692e37c4b4afc60ec13bb0b99886f129bf5e85ac976fb1913b623e449fd5aCOPY
scripts /workspace/nemo/scripts
02/27/2024 5:37 AM UTC
6b66d35f533fd5c976290bdc953d41563cdcf42408124926c7486c5289885672WORKDIR
/workspace/nemo
02/27/2024 5:37 AM UTC
14bf5217dde37ff3e360bff1f0b50c561c3a654649fb2d4578908a37efff3a5dRUN
NEMO_VERSION=1.23.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"
02/27/2024 5:37 AM UTC
047bb7e07b6336f2bdbc5fb6cda9783c5b273438ae18dbdff88cb78ffc22f36aRUN
NEMO_VERSION=1.23.0 cd /tmp/nemo &&
  pip install ".[all]"
02/27/2024 5:36 AM UTC
b02f7ac4030c3a67d9dbc4d7bfe82e4d25920e862db2fcab02aa70d48c5db441RUN
NEMO_VERSION=1.23.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
02/26/2024 10:46 PM UTC
...