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
df2e21615fe723966f0666b0d3558b8d30fac34d4bc766b48608898ee4a63f91CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
12/09/2021 2:10 AM UTC
b6a1f5ff3b18861035613ab77061fd6679519cf3efcd7152dc120c8447fe5844COPY
tutorials /workspace/nemo/tutorials
12/09/2021 2:10 AM UTC
4e9798e46851bdcd50547793902ca9fc6ae20134bd6109af17bd2637c243af0dCOPY
tests /workspace/nemo/tests
12/09/2021 2:10 AM UTC
ee7343425767c30fecd5fcd0e16ae23ada21b7ebd059354ae1120d74bf429f1dCOPY
examples /workspace/nemo/examples
12/09/2021 2:10 AM UTC
9c9dde0e420c53df44c72fc40b076ce193bad2ae9d7fa9055afe4263a7922b26COPY
scripts /workspace/nemo/scripts
12/09/2021 2:10 AM UTC
9252cda28b02cb7d61fa216a3652811cc07910a817e0135cc7c805d933216454WORKDIR
/workspace/nemo
12/09/2021 2:10 AM UTC
f7b922782ef3657e2f05ec2fd07118027804df79c4bd9af963e4e41a7bcbe1d1RUN
NEMO_VERSION=1.5.1 rm -rf /opt/conda/pkgs/conda-4.11.0-py38h578d9bd_0/info/test/tests/data/env_metadata/
12/09/2021 2:10 AM UTC
7ef75dd4283c4c4091a0372a904a229b22df007fc9069e8cc580e78fc0f0a797RUN
NEMO_VERSION=1.5.1 conda install -c conda-forge numba=0.54.1
12/09/2021 2:10 AM UTC
f2ac05bfa2edba66c55b4b3697fa65f4804400f6521ce08189150b64e4df88f3RUN
NEMO_VERSION=1.5.1 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"
12/09/2021 2:10 AM UTC
b17ef91d93b2d6701f727812444264a12c1385f106c3e9e8bee97413c0720cbaRUN
NEMO_VERSION=1.5.1 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
12/09/2021 2:09 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.