NeMo Framework Megatron Backend

NVIDIA

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.

NVIDIA AI Enterprise Supported

Layer	Label		Created
sha256:1aea5b6b294f854dc7d1aebec419749a857aff0de84049741394f4fa7b84feec	RUN	`NEMO_VERSION=1.9.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh && chmod +x start-jupyter.sh`	06/06/2022 5:46 PM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1	RUN	`NEMO_VERSION=1.9.0 rm -rf /opt/conda/pkgs/conda-4.11.0-py38h578d9bd_0/info/test/tests/data/env_metadata/`	06/06/2022 5:46 PM UTC
sha256:ff6e8bce7b5cefae160cca236310776b2a0abf567fb4ba6ced8bd74fa2e92a38	COPY	`tutorials /workspace/nemo/tutorials`	06/06/2022 5:46 PM UTC
sha256:c5341470684c2a916e383dd634f0f623d8a1a557852cb5b4415708fd471cac62	COPY	`tests /workspace/nemo/tests`	06/06/2022 5:46 PM UTC
sha256:3f2348cd959e0f9a665c2a533c2d0275e88ed122c4a550e0794b64314294da89	COPY	`examples /workspace/nemo/examples`	06/06/2022 5:46 PM UTC
sha256:0c54357a68492b419048d7763990fc01ba372da0ca37249f176cc8135e964782	COPY	`scripts /workspace/nemo/scripts`	06/06/2022 5:46 PM UTC
sha256:bd08a81fcdefca534594f2554e24c56406e45759525adf1eed50d8a3111cff39	WORKDIR	`/workspace/nemo`	06/06/2022 5:46 PM UTC
sha256:d2acc7a7c7a2621261387743494f39384c840c024996ae40993e369a88795da7	RUN	`NEMO_VERSION=1.9.0 cd /tmp/nemo && pip install ".[all]" && sed -i "s/_module_available(\"torchtext\")/False/g" /opt/conda/lib/python3.8/site-packages/pytorch_lightning/utilities/imports.py && 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/06/2022 5:46 PM UTC
sha256:63caa352fefab1a0ee1af6c1e3a7f9d283ed730c2a3d1f52309238be3aa7ccd3	RUN	`NEMO_VERSION=1.9.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`	06/06/2022 5:45 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`NEMO_VERSION=1.9.0`	06/06/2022 5:45 PM UTC