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:1653678f004915283fbe529bfb8b780d1c0760d31d3fc34af24c7130fe882554RUN
    NEMO_VERSION=v1.0.0b2-2-g93a3d8b7 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
      chmod +x start-jupyter.sh
    12/11/2020 7:06 PM UTC
    sha256:891d93bf347af7985086eda8fa01400269cdd20444383ed2e26e549e9216ed6eCOPY
    tutorials /workspace/nemo/tutorials
    12/11/2020 7:06 PM UTC
    sha256:2fd1116757e1312e0a2ab0a40702c7f7fd207e9e136424604231099612651562COPY
    tests /workspace/nemo/tests
    12/11/2020 7:06 PM UTC
    sha256:f803882bf3ceffae6b266845bfa4525f04249b136ff48173b4b86521d10a5916COPY
    examples /workspace/nemo/examples
    12/11/2020 7:06 PM UTC
    sha256:2dc67d875595ce6bd32e6fa4ad5bd894ef7e776143fb0dd02319630a997157b1COPY
    scripts /workspace/nemo/scripts
    12/11/2020 7:06 PM UTC
    sha256:3abd8f09493335251c11f85e0514248576694cba6fee5e6ce1b2a8cdfafcde1eWORKDIR
    /workspace/nemo
    12/11/2020 7:06 PM UTC
    sha256:72cd48e620462ea9ab5fa1771ba1edea410a51a3b851f68dd5ac9db189f0c7ecRUN
    NEMO_VERSION=v1.0.0b2-2-g93a3d8b7 cd /tmp/nemo &&
      pip install ".[all]"
    12/11/2020 7:06 PM UTC
    sha256:bca5d80e5ee1cb2aacdfbaf3b68b04824fc7059afd2c985f8402d5b15c502ae9RUN
    NEMO_VERSION=v1.0.0b2-2-g93a3d8b7 /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/11/2020 7:05 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NEMO_VERSION=1.0.0b3
    12/11/2020 1:28 AM UTC
    sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1RUN
    for f in $(ls requirements/*.txt); do pip install --disable-pip-version-check --no-cache-dir -r $f; done
    12/11/2020 1:28 AM UTC
    ...