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
54bcae0c75f7eaad0511c12e0032b4db997046a4f3d788a0ae1409bc5eb5f763CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
12/09/2022 6:38 AM UTC
8a79ba763cfeda8f07aba797c86e99ad586f989eaac791dea72e4313baa836e2RUN
NEMO_VERSION=1.13.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
12/09/2022 6:38 AM UTC
572b79f7be492ea80a7ffc7d292728ccfa480ebe721094349ae78f937acbbcf6COPY
tutorials /workspace/nemo/tutorials
12/09/2022 6:38 AM UTC
be083fa5ec68c123bca2c4c8eda56517de9966eac0c19bd379eefa9cecfb515fCOPY
tests /workspace/nemo/tests
12/09/2022 6:38 AM UTC
37fea8bc04cd0895ab2efb0dd6663a0cea56ea91e5e01586849773c683a89234COPY
examples /workspace/nemo/examples
12/09/2022 6:38 AM UTC
262c864440ac2491936e7fa505ab3189e51fdef2ae28ba4279ffa9ecef004b2cCOPY
scripts /workspace/nemo/scripts
12/09/2022 6:38 AM UTC
eabe8be0fc0e0e0963f5b0801ab5719b97bf869304b704e8f67a08f4d2c72c33WORKDIR
/workspace/nemo
12/09/2022 6:38 AM UTC
bb410b8df07703db289b16bc4cd9a6bcefe8c5a26d5ce3d35c4c60e38acce1beRUN
NEMO_VERSION=1.13.0 cd /tmp/nemo &&
  pip install ".[all]" &&
  sed -i "s/_module_available(\"torchtext\")/False/g" /opt/conda/lib/python3.8/site-packages/pytorch_lightning/utilities/imports.py &&
  python -c "import nemo.collections.asr as nemo_asr" &&
  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"
12/09/2022 6:38 AM UTC
34e4c1f7f9d7214e4d1934b51c827de620ac31d9c94d50d0654827bef66c3497RUN
NEMO_VERSION=1.13.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
12/09/2022 6:38 AM UTC
3a547b33f1f54f98b0562bdbd427c28a5aaf7c1601cbcc25119c7514975f4b7bARG
NEMO_VERSION=1.13.0
12/09/2022 6:37 AM UTC
...