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: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
...