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
6b65f4b3983817905930c0ef8c2211dd8614dd3d0ee9687f837028d0ea35e1e9CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
04/05/2023 8:02 PM UTC
47d2ac17c0cae9a50e73119e3d1101e854ea4a2dab8acf7da2dacb8e3a4c7d74RUN
NEMO_VERSION=1.17.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
04/05/2023 8:02 PM UTC
102d8271428d236c8160759a1be9d081d80cec56a57459dae5e906cb08368fdcCOPY
tutorials /workspace/nemo/tutorials
04/05/2023 8:02 PM UTC
28bd1470537232d24b61fffc1d3339d1d12df9ef769a01932d42ac3f53910c1aCOPY
tests /workspace/nemo/tests
04/05/2023 8:02 PM UTC
8a426b6db2c41debd9535b8f9316c66225c1e10a77494e6fcb17f5865a7dc43bCOPY
examples /workspace/nemo/examples
04/05/2023 8:02 PM UTC
43071255401e401f8fa4eafcda7796fc2b7085710f447a1865d796914e62ddfaCOPY
scripts /workspace/nemo/scripts
04/05/2023 8:02 PM UTC
8f9f44f2fd3febd77f043e717f1c63f0b47ec732083b2cadc51432c538dfed7cWORKDIR
/workspace/nemo
04/05/2023 8:02 PM UTC
8d6a1e507a08c88a183a1f58ae99e3ff6d51dfdfbd51d6244b9a5988c765635cRUN
NEMO_VERSION=1.17.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"
04/05/2023 8:02 PM UTC
cf5cb4350a746209d172e50baec5894dc1aa020d76bc07c599128f52f4e4a9dfRUN
NEMO_VERSION=1.17.0 cd /tmp/nemo &&
  pip install ".[all]"
04/05/2023 8:02 PM UTC
3a292fb1aced711ef6319063499051faf1a01428a7be7c441c135f6fce6d9c1bRUN
NEMO_VERSION=1.17.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
04/05/2023 8:02 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.