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
b72a5f9929ddcbd49c79d5cf4ab0691deb9f1ee000778bdaaef470d99de9d3cfCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
10/26/2023 10:21 PM UTC
0701563f7d93883a43fef632402a2b63b77ee0f01c2b657cb521fe989bb7739eRUN
NEMO_VERSION=1.21.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
10/26/2023 10:21 PM UTC
2956ddfc8dcc9aee51a9ae0564a876ef0de8d9cacada62e45bd712bd90bb4ff0COPY
tutorials /workspace/nemo/tutorials
10/26/2023 10:21 PM UTC
9a661f0093f835d6dee359beb89f7dcda7389e32bec3911f386d8a4cfe49079bCOPY
tests /workspace/nemo/tests
10/26/2023 10:21 PM UTC
dbad51739c781e16063d317ddb47ad970e346db001a18c4abe57c63060209249COPY
examples /workspace/nemo/examples
10/26/2023 10:21 PM UTC
cd87ad8d5ec8122a518f350529590005a9b01c7b34d90565292ad362614947e2COPY
scripts /workspace/nemo/scripts
10/26/2023 10:21 PM UTC
84f41d984ce3352b04287eca1f3a5875fc72e0c7a7c23ccdd900ce2c0f1a54a2WORKDIR
/workspace/nemo
10/26/2023 10:21 PM UTC
5007772350dae7e79e7cd06955219d4421ca8fcbeefae7649683db4d5af34467RUN
NEMO_VERSION=1.21.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"
10/26/2023 10:21 PM UTC
ff30ea0da99a250f2e69f6e7e904711df9822d5401f0152764af4d561f834cc4RUN
NEMO_VERSION=1.21.0 cd /tmp/nemo &&
  pip install ".[all]"
10/26/2023 10:21 PM UTC
5b67d191fa8a88c99ece521f72d09e89e3e98a2640c54c16adeb541b73ba4348RUN
NEMO_VERSION=1.21.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
10/26/2023 10:20 PM UTC
...

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.