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
b0ac217b80ba0bc4f6179a5477747563a268d04ae62d8c16b6850b26b48ec3f1CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
05/17/2023 10:29 PM UTC
32050d7a8a1df8b47e7f793220f2c6e6daed3ee5aa6468ad7a5157c41253962eRUN
NEMO_VERSION=1.18.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
05/17/2023 10:29 PM UTC
7194d594f72199af0e5542b6bec635ea9f5fc778919139dd37c1dd2814b1ae53COPY
tutorials /workspace/nemo/tutorials
05/17/2023 10:29 PM UTC
1ed0cc53b4c21378290c3a68b15dd298028bdac33ea0064f4b79bf2091a41e15COPY
tests /workspace/nemo/tests
05/17/2023 10:29 PM UTC
14d517b782ebf1ce6c1aa752fcb8de9304652933dd2ac512b0c5c8cfdcc0429fCOPY
examples /workspace/nemo/examples
05/17/2023 10:29 PM UTC
c75c7283964dc48379fd73b51eb8e69eccea0afe54a75498610cc6b034708fbaCOPY
scripts /workspace/nemo/scripts
05/17/2023 10:29 PM UTC
61cae4c3625b8e89b5f234e19a913a8c5547af4d42a293f8ee3d9ac8c5a25b86WORKDIR
/workspace/nemo
05/17/2023 10:29 PM UTC
8d35aa08d2bd4dd6092f9618b6c5e4e706062a11e1313cb3a42d4a8a84c56c23RUN
NEMO_VERSION=1.18.0 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"
05/17/2023 10:29 PM UTC
e2c6551c9e84568a08dd2f87f75ed1f460dd8ace3ed91b6511d22be19022b3f2RUN
NEMO_VERSION=1.18.0 cd /tmp/nemo &&
  pip install ".[all]"
05/17/2023 10:29 PM UTC
75a13dcac724af5009dfb6d1c236c10bcad3d3f46d3a0b6a791a6ece01811df4RUN
NEMO_VERSION=1.18.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
05/12/2023 7:28 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.