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.

  • LayerLabelCreated
    sha256:efe51a222638855e0c5c4a8d135b94930b1299e871ccdddd6fc351a99531815fRUN
    NEMO_VERSION=stable-76-g624099bd printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
      chmod +x start-jupyter.sh
    11/16/2020 10:42 PM UTC
    sha256:f5c5088ffd25e142a557d18435a1b52d223507f5d5fa44a5ffba43da69851d96COPY
    tutorials /workspace/nemo/tutorials
    11/16/2020 10:42 PM UTC
    sha256:5a263be31f632e3938486d98db36ec7801041b53e59ea3f4dda5e9885255df92COPY
    tests /workspace/nemo/tests
    11/16/2020 10:42 PM UTC
    sha256:60a758e8599a0a508a2d2db1433a0f6fbe9ccc35dc498efb010d4a94c644b7c1COPY
    examples /workspace/nemo/examples
    11/16/2020 10:42 PM UTC
    sha256:5ac7c9262d6d437097f2df47ec6e2584d97af76cb1b3a4077c92725f8465a261COPY
    scripts /workspace/nemo/scripts
    11/16/2020 10:42 PM UTC
    sha256:1e2986c68e0f3f5a8a087ba42fff9e8307d892ebed9bfc2bc1bf7567ee35a5d2WORKDIR
    /workspace/nemo
    11/16/2020 10:42 PM UTC
    sha256:292b71c61d57a9fe65dc2856970ecabc050ab05937608c1ac7e71c13bae9542bRUN
    NEMO_VERSION=stable-76-g624099bd cd /tmp/nemo &&
      pip install ".[all]"
    11/16/2020 10:42 PM UTC
    sha256:5b92e53e5c203466aeb9ae106d1cd3ad7e41b825fb5be40a41b9315e835e5245RUN
    NEMO_VERSION=stable-76-g624099bd /usr/bin/test -n "$NEMO_VERSION" &&
      /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
      /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
    11/16/2020 10:42 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NEMO_VERSION=1.0.0b2
    11/16/2020 10:42 PM UTC
    sha256:fb90cf67839303394c28c796bdca81439ff30f5abdddd37d564dc8e1696b5f7cRUN
    ./install_ctc_decoders.sh
    11/16/2020 10:42 PM UTC
    ...