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:6d47becb4213273f9e7e5555a82b6cd1bdffb74f23d9c7fa119fa716c6811a04RUN
NEMO_VERSION=1.0.0b1 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
10/05/2020 5:06 PM UTC
sha256:3c7f7fc54d30465c4ceb844fe15c282e8db508432d0eede7c588b37e864cece2COPY
tutorials /workspace/nemo/tutorials
10/05/2020 5:06 PM UTC
sha256:70b55f78432080d5ff3ae83ffaecfbfcaf11720d33d807c99ba9a29a32108460COPY
tests /workspace/nemo/tests
10/05/2020 5:06 PM UTC
sha256:752d9d56ed886d80f8a5ef6b5cde538af52cad96394ae7cd0f8d267e34adbf45COPY
examples /workspace/nemo/examples
10/05/2020 5:06 PM UTC
sha256:bb655e0992ad71e6dc5baacfef21d14c8fa10d19cec7fa2fc87bf9f436b96b89COPY
scripts /workspace/nemo/scripts
10/05/2020 5:06 PM UTC
sha256:8a470eb774223310475cc54d2ca87131d45c59066ec3a4f1036bcfa902a9f736WORKDIR
/workspace/nemo
10/05/2020 5:06 PM UTC
sha256:194d366de06ff69bb35cb766e47ed3bf99ae6ae6afb4c019f716a1e655851956RUN
NEMO_VERSION=1.0.0b1 cd /tmp/nemo &&
  pip install ".[all]"
10/05/2020 5:06 PM UTC
sha256:0eb8b585748d0c8e9149a54a156ce8706551d29cc3f52cb9c8678d0e44079da4RUN
NEMO_VERSION=1.0.0b1 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
10/05/2020 4:59 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NEMO_VERSION=1.0.0b1
10/05/2020 4:59 PM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1RUN
for f in $(ls requirements/*.txt); do pip install --disable-pip-version-check --no-cache-dir -r $f; done
10/05/2020 4:59 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.