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
sha256:5a09f4d2ee1a4eb82754bb8dd48037ca4297639b7a33a0ae4fa8138d5ad601a0RUN
NEMO_VERSION=1.0.2 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
07/30/2021 10:51 PM UTC
sha256:9725c4d84a8fa2051bdef2e2f733dce78892c37f878147a8cf6411904a80ab8aCOPY
tutorials /workspace/nemo/tutorials
07/30/2021 10:51 PM UTC
sha256:1953274fd4677fa1b9c074fa14853f982e3fe59a28f8060c2d1dcc7ead0ced53COPY
tests /workspace/nemo/tests
07/30/2021 10:51 PM UTC
sha256:af6c1b1a92fe6dffbb2c055417264bd4758209c8f43d5990964b00cb42d52335COPY
examples /workspace/nemo/examples
07/30/2021 10:51 PM UTC
sha256:ffe5b10cc4fca2d2abf3581b0e58be98d10e7c771bb938abadbd477db7a30506COPY
scripts /workspace/nemo/scripts
07/30/2021 10:51 PM UTC
sha256:7fb37d981be9015ca2ec52277497c49b1a64981c4df7dc6e9dc7a45fd9a2d656WORKDIR
/workspace/nemo
07/30/2021 10:51 PM UTC
sha256:2ecfceaa1e04eef472fe27089e4b464d02cf5065ff233a0ad21148fb8a148c35RUN
NEMO_VERSION=1.0.2 conda install -c numba numba=0.53.1
07/30/2021 10:51 PM UTC
sha256:4bc121e1249d768fdfb16ad34e778988bff7c1965aa719a041b359ccc8d13012RUN
NEMO_VERSION=1.0.2 cd /tmp/nemo &&
  pip install ".[all]" &&
  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"
07/30/2021 10:50 PM UTC
sha256:3cf82fcbbc30bba6b4062c57c253da386bb582b490626e8d312aa2fac559c04fRUN
NEMO_VERSION=1.0.2 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
07/30/2021 10:50 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NEMO_VERSION=1.0.2
07/30/2021 10:50 PM UTC
...