Skip to main content
NVIDIA
NeMo Framework Megatron Backend
Container
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:9d1dfdc02ad7b778eec2c319226e691dc3710e56abc0abc1d43008ee1dc2f21cRUN
    NEMO_VERSION=1.3.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
      chmod +x start-jupyter.sh
    09/03/2021 10:45 PM UTC
    sha256:5ab55d3a5ab264e1daea010ac85bedf3630910c8ebca608a197880531a69a404COPY
    tutorials /workspace/nemo/tutorials
    09/03/2021 10:45 PM UTC
    sha256:e81081f3e13f056bf353a9cc447f4b045a9598b49167907e79d464180bc5e20fCOPY
    tests /workspace/nemo/tests
    09/03/2021 10:45 PM UTC
    sha256:66f583cbd6cbe9591fb7e0e95560eb0cc7d3874dbac6cf2455149d1c3feefbf8COPY
    examples /workspace/nemo/examples
    09/03/2021 10:45 PM UTC
    sha256:85696c3196a72080f21ae99ff6e3ff9f52e413225a654bb4649742fd2e8bafe5COPY
    scripts /workspace/nemo/scripts
    09/03/2021 10:45 PM UTC
    sha256:162ece3c0c7813550682030c4dced5d8448277f8b345aa1e4c084f737d7e4c55WORKDIR
    /workspace/nemo
    09/03/2021 10:45 PM UTC
    sha256:5cd1d9dc936cdd335e0da2b3c8cad5b2e1edc815f1d72ecabca27fc270177183RUN
    NEMO_VERSION=1.3.0 conda install -c numba numba=0.53.1
    09/03/2021 10:45 PM UTC
    sha256:3f9ab7ec40ae680ab582f06294b2da8abd6fbb360dbc10390979cac7ed34bbf9RUN
    NEMO_VERSION=1.3.0 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"
    09/03/2021 10:44 PM UTC
    sha256:047db7e6c8bcd23af8625a2a531c613c4b3683c680586d6d56e57fde225dbac9RUN
    NEMO_VERSION=1.3.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
    09/03/2021 10:44 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NEMO_VERSION=1.3.0
    09/03/2021 10:44 PM UTC
    ...