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:2af37c06a86a5f9c5bf5ddcf063c1a006a39665b5ce9b49b7b89015905e4e85aRUN
NEMO_VERSION=1.0.1 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
06/10/2021 12:56 AM UTC
sha256:d1cad126274859e0014f3aa049e2b09a5fb7b72156358fd25ce12723b6bc34e5COPY
tutorials /workspace/nemo/tutorials
06/10/2021 12:56 AM UTC
sha256:ce3ad68ef70eee3abb02d7bf1159ed613b9472ed4fdd3be4d1ff3ef59df0d372COPY
tests /workspace/nemo/tests
06/10/2021 12:56 AM UTC
sha256:cd39e801cd0bb48521590bea86e6d7ce2a72733aff57c065c258b17c5eecfcf5COPY
examples /workspace/nemo/examples
06/10/2021 12:56 AM UTC
sha256:b47da202a6cfbe1edf1594a9e2a4db87786c4adbc6cf97ea0c928e75b3910aabCOPY
scripts /workspace/nemo/scripts
06/10/2021 12:56 AM UTC
sha256:cbb7eb53e0e29a44a6e73dcb79905d4146df3731035ef293ad007fd2cf436c37WORKDIR
/workspace/nemo
06/10/2021 12:56 AM UTC
sha256:c3d864cd2c2ef26a046f920f3fc8ca4b7937cd72c70e5b95c3e0caaa45e8dcd1RUN
NEMO_VERSION=1.0.1 conda update -c numba numba -y &&
  pip install --upgrade PyYAML
06/10/2021 12:56 AM UTC
sha256:2315bb6bef8d3d6704a7d72a131d99c888c00ecd8192720f60497a25b2d2017eRUN
NEMO_VERSION=1.0.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"
06/10/2021 12:55 AM UTC
sha256:b9877329d7775bba55041485ec9827faab7e335ba44b2e0e38c633a4248f36baRUN
NEMO_VERSION=1.0.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
06/10/2021 12:55 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NEMO_VERSION=1.0.1
06/10/2021 12:55 AM UTC
...