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
c831c2d0ede7de55efdf36728b187c89c5edc71a7c6d13208e6506dc44652bfeCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
10/11/2022 9:07 PM UTC
5c9830a06dafc9984d4f7e3dad93b73139df42f197e44a112ae7938bb6fbe9caCOPY
tutorials /workspace/nemo/tutorials
10/11/2022 9:07 PM UTC
e5af65aeab305461fb32b59cfa6c59adae04a83d087be418c8464fabd6b8558dCOPY
tests /workspace/nemo/tests
10/11/2022 9:07 PM UTC
e97437f41d8bea82b24b45b99f2c23a29b103e82b80527e64feef8b14034bcb1COPY
examples /workspace/nemo/examples
10/11/2022 9:07 PM UTC
054b0f36ef1082eca384647f2c2d851942b932af7336a2cef641d6714d8ebe83COPY
scripts /workspace/nemo/scripts
10/11/2022 9:07 PM UTC
a3c7114d623130a81dc86d043faa4d52dc0652e3c28c8d90d6e1fa6b6012d643WORKDIR
/workspace/nemo
10/11/2022 9:07 PM UTC
90f9ed46c914d558587a3541a315bc264399c2c2939861ac82ca385d3791ff27RUN
NEMO_VERSION=1.12.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"
10/11/2022 9:07 PM UTC
aa64864314a18515a529fb5ba6373e5355e08ee3c839858e96cd308e57d5c393RUN
NEMO_VERSION=1.12.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/11/2022 9:06 PM UTC
2c7d1de8c6d3e0d3d4488c728b9379d136f19d8f99afba7d79280510b6861a7bARG
NEMO_VERSION=1.12.0
10/11/2022 9:06 PM UTC
a10d4a8ceb22b1c59738442b44f31657fec648e9cb46de01072dc612475376efRUN
/bin/bash /tmp/nemo/scripts/speech_recognition/k2/setup.sh ||
  exit 0
10/11/2022 9:06 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.