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
500e2a71ca36e504ad10a0fcf5b2e4345ba56c97105c692e9cf1e1e3976a45dbRUN
/bin/sh
01/02/2020 6:35 PM UTC
acb65b6474e2110bf8159ae652f6cb31124c086d2012017b0ac077202662ed19RUN
printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
01/02/2020 6:35 PM UTC
998eaa3d8c0c1472481c01f3c3a65b6e49c630022f798397dd6a79e84ccfc169RUN
mv examples/ /workspace/nemo_examples &&
  rm -rf /workspace/nemo
01/02/2020 6:35 PM UTC
cf7ea0d18179c665aa2dbd1b680c9495b5503521906c3ab36e525e5e4bb6dcfeWORKDIR
/workspace/nemo
01/02/2020 6:34 PM UTC
1e54c83931b995ff722bdf2ab3d06f6a8b7e16208c96d316ffa027fc3c489746RUN
cd nemo &&
  pip install . &&
  cd ../collections/nemo_asr &&
  pip install . &&
  cd ../nemo_nlp &&
  pip install . &&
  cd ../nemo_simple_gan &&
  pip install . &&
  cd ../nemo_tts &&
  pip install .
01/02/2020 6:34 PM UTC
69b7a27ae8427ef7919be6af76a281f3e4418f31718b3bcf1aa32540beca8524COPY
dir:bd5d0adc28f5127bd0193eac1ea9433f2281da7a24bfe029cbd9f9c41fc98040 in .
01/02/2020 6:34 PM UTC
e304d5e297765823a48d9e0d6b7a0d69ce5b2a813fdf4084fe3dfb634e5b84dbRUN
pip install --disable-pip-version-check -U -r requirements.txt
12/07/2019 12:08 AM UTC
e104fdcfef3eec641e570e986a4657d478a74443c0ea3e9b62d7b358074f84c3COPY
file:133daf3065d6e3ce2917f4aa30f7a36dc743a7331f787c4c87306ccc50d4af78 in .
12/07/2019 12:07 AM UTC
2d79cd585a1a84bafa6077258883d53cbbd1a0bf16519d565f625f7cd469b714WORKDIR
/workspace/nemo
12/07/2019 12:07 AM UTC
280f92b4cedcb355897cc04ba185f5561d0a9615ac06e5a33cd96cd040d6eb0dRUN
git clone https://github.com/onnx/onnx-tensorrt.git &&
  cd onnx-tensorrt &&
  git submodule update --init --recursive &&
  patch -f < ../onnx-trt.patch &&
  mkdir build &&
  cd build &&
  cmake .. -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=/usr -DGPU_ARCHS="60 70 75" &&
  make -j16 &&
  make install &&
  mv -f /usr/lib/libnvonnx* /usr/lib/x86_64-linux-gnu/ &&
  ldconfig
12/07/2019 12:07 AM 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.