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
a95bed609c369a0bb13bd36e62bed584f46484300e7abadeca17482d7ed60dcaCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
08/02/2023 10:58 PM UTC
1fb8ef5ec02e0bb40f9ce30318a65cf9b2719f241f24d910b138877a3aaf427eRUN
NEMO_VERSION=1.20.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
08/02/2023 10:58 PM UTC
ba375dae69911fc5fbcb45ec45233bef1fe834277431c404ac84d9d3d52d4929COPY
tutorials /workspace/nemo/tutorials
08/02/2023 10:58 PM UTC
11378c7586527d9045cce8dd669a19207dac6371dde04bd3bcab59df86cd650cCOPY
tests /workspace/nemo/tests
08/02/2023 10:58 PM UTC
8c60bae643ada26fc530bac3493d371e63cc4cc2be44ce4db8d96b5fcb9d10c8COPY
examples /workspace/nemo/examples
08/02/2023 10:58 PM UTC
179b5ab71ac53c138893837d9374ed97eb7ef0c623315f43ba83e1543c2383daCOPY
scripts /workspace/nemo/scripts
08/02/2023 10:58 PM UTC
c53782c93f24d0983e1b391ae2480d3d2040f61866b87a2f2cf6aba4df820078WORKDIR
/workspace/nemo
08/02/2023 10:58 PM UTC
ac5e44dc4f0923ebd2cfdaef42408e428bd65ec63f5d98391ffff374bc39c657RUN
NEMO_VERSION=1.20.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"
08/02/2023 10:58 PM UTC
2f0642cbc32ad6545f8564dc37674e0880459f8d69d33a3d21d3935425bd8e9dRUN
NEMO_VERSION=1.20.0 cd /tmp/nemo &&
  pip install ".[all]"
08/02/2023 10:57 PM UTC
456b980fd4d1d16a2bd2a745d294099ce89316dd8de34c8923e5dd0b758b4239RUN
NEMO_VERSION=1.20.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
07/25/2023 1:32 AM 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.