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
c5e665357398604b36dc8e3ea8c6f9b2df9e0c066bda4a723de64c9c5f194afaCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
06/06/2022 5:46 PM UTC
2f06a2965c84e260dd861e746b2a7bab9b6b5f8b74407bcf23dc22ccf0f8566eRUN
NEMO_VERSION=1.9.0 rm -rf /opt/conda/pkgs/conda-4.11.0-py38h578d9bd_0/info/test/tests/data/env_metadata/
06/06/2022 5:46 PM UTC
c47480a778f7f36f9abc25c465270757f01e35ae8687c5f0f7d771575070bd0aCOPY
tutorials /workspace/nemo/tutorials
06/06/2022 5:46 PM UTC
e76d8222ae913b51cf05a8da4db1e70f7da4bbaeec50b517d0003d82b28b624fCOPY
tests /workspace/nemo/tests
06/06/2022 5:46 PM UTC
71f3cf50c1a21650a68fde8d61393cb167443285521b1b9501b5bfedd5cafaabCOPY
examples /workspace/nemo/examples
06/06/2022 5:46 PM UTC
a64ef6c765789a1a15b4505ddc69926b339bd85d3a337aa5e256e2e793f9dc5bCOPY
scripts /workspace/nemo/scripts
06/06/2022 5:46 PM UTC
6ab134d4e1b26777b2f391a17801f1da60e1b5a99dab3e1fa8f93607927241fcWORKDIR
/workspace/nemo
06/06/2022 5:46 PM UTC
f9f7c10dc2af928e81c527fa56db80fd7ada043b0c675caa12cef4778d0d9196RUN
NEMO_VERSION=1.9.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"
06/06/2022 5:46 PM UTC
9a3c645005bf1c0b29416d4c26dee18312b3da6f4fb97084aac75a231d59725dRUN
NEMO_VERSION=1.9.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
06/06/2022 5:45 PM UTC
8afc40ba465e29eb4b86e0090b6f49aff9c9109b4ff1995fdffc62c1768b4d32ARG
NEMO_VERSION=1.9.0
06/06/2022 5:45 PM UTC
...