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
a02d3cf6820633fe079ff0075cef5b1c435c86c7f7cc62f31539f7bf8c4054c8CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
02/01/2023 6:52 PM UTC
cb0e057d9172ed7beaf1daf6cef1d1803e23f8e447488fb645e3c238b129bcb7RUN
NEMO_VERSION=1.15.0 AIS_VERSION=v1.3.15 AIS_BIN=https://github.com/NVIDIA/aistore/releases/download/v1.3.15/ais-linux-amd64.tar.gz curl -LO ${AIS_BIN} &&
  tar -xzvf ais-linux-amd64.tar.gz &&
  mv ./ais /usr/local/bin/. &&
  rm ais-linux-amd64.tar.gz
02/01/2023 6:52 PM UTC
b04ab1d8b1c7e41c73ec7f4daf33d3ea947126851566fe6513ccde6276856727ARG
AIS_BIN=https://github.com/NVIDIA/aistore/releases/download/v1.3.15/ais-linux-amd64.tar.gz
02/01/2023 6:52 PM UTC
df57959cb0f4bc5e288692ee0eb289f2dd551aebb5f8411b956749dedbf08d9dARG
AIS_VERSION=v1.3.15
02/01/2023 6:52 PM UTC
b63b68cb86db0d65eef9b9fdf7ed24c61de7f79d6e6705f9c648ee604c35ab26RUN
NEMO_VERSION=1.15.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
02/01/2023 6:52 PM UTC
f84487fe63bb3ae7f72d4b4c96fb4468d06a73f2639a4a0fd204b6503b5fa49aCOPY
tutorials /workspace/nemo/tutorials
02/01/2023 6:52 PM UTC
25ea7103788f3bce22bb0f825ab3be34236bb2e8fc5af220c025af0e48abb1b8COPY
tests /workspace/nemo/tests
02/01/2023 6:52 PM UTC
aa6274b5bbad9c9c59590f362c0d4b83ec93b206dca29bb4aa3d16f59170f674COPY
examples /workspace/nemo/examples
02/01/2023 6:52 PM UTC
91c54fa89abc1fe0bccee389c7a2b4304a81ecd9a03dfb84a5e401bb823aa04cCOPY
scripts /workspace/nemo/scripts
02/01/2023 6:52 PM UTC
e7d0fa2868d9a792affeac9170f7bb2378015731a92865ba21bf600313440198WORKDIR
/workspace/nemo
02/01/2023 6:52 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.