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:efe51a222638855e0c5c4a8d135b94930b1299e871ccdddd6fc351a99531815fRUN
NEMO_VERSION=stable-76-g624099bd printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
11/16/2020 10:42 PM UTC
sha256:f5c5088ffd25e142a557d18435a1b52d223507f5d5fa44a5ffba43da69851d96COPY
tutorials /workspace/nemo/tutorials
11/16/2020 10:42 PM UTC
sha256:5a263be31f632e3938486d98db36ec7801041b53e59ea3f4dda5e9885255df92COPY
tests /workspace/nemo/tests
11/16/2020 10:42 PM UTC
sha256:60a758e8599a0a508a2d2db1433a0f6fbe9ccc35dc498efb010d4a94c644b7c1COPY
examples /workspace/nemo/examples
11/16/2020 10:42 PM UTC
sha256:5ac7c9262d6d437097f2df47ec6e2584d97af76cb1b3a4077c92725f8465a261COPY
scripts /workspace/nemo/scripts
11/16/2020 10:42 PM UTC
sha256:1e2986c68e0f3f5a8a087ba42fff9e8307d892ebed9bfc2bc1bf7567ee35a5d2WORKDIR
/workspace/nemo
11/16/2020 10:42 PM UTC
sha256:292b71c61d57a9fe65dc2856970ecabc050ab05937608c1ac7e71c13bae9542bRUN
NEMO_VERSION=stable-76-g624099bd cd /tmp/nemo &&
  pip install ".[all]"
11/16/2020 10:42 PM UTC
sha256:5b92e53e5c203466aeb9ae106d1cd3ad7e41b825fb5be40a41b9315e835e5245RUN
NEMO_VERSION=stable-76-g624099bd /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
11/16/2020 10:42 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NEMO_VERSION=1.0.0b2
11/16/2020 10:42 PM UTC
sha256:fb90cf67839303394c28c796bdca81439ff30f5abdddd37d564dc8e1696b5f7cRUN
./install_ctc_decoders.sh
11/16/2020 10:42 PM UTC
...

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.