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
05ebb81229aef6ef0a484203a4595d40366d82b58729e3f9466dfaf00b5ed3f5CONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
06/10/2021 12:56 AM UTC
24db57b86fa96dc61d6bb88dc8837dbd9dca16789fe4f03aa194caddc63664e7COPY
tutorials /workspace/nemo/tutorials
06/10/2021 12:56 AM UTC
d3444c8ca15f8a513666d1386eb7386ac3580e22481891ef1ddcf95a8281e1b1COPY
tests /workspace/nemo/tests
06/10/2021 12:56 AM UTC
aae793ef71df4c73c42ca3fafe1eb187b26dfc6969f9495e89ba388e6e06e833COPY
examples /workspace/nemo/examples
06/10/2021 12:56 AM UTC
ded9b53054e34b5087e61c25891d9e8df4b9549b6e750ae2145f1abb1faad898COPY
scripts /workspace/nemo/scripts
06/10/2021 12:56 AM UTC
b590436c31a006d7fb21fa483a07d04c97214f232672e2379e86bfc763732fefWORKDIR
/workspace/nemo
06/10/2021 12:56 AM UTC
b8b855c6ee9bf4fcc0e507782015555e47f76913c2d3ba90c32ebf05027d2a68RUN
NEMO_VERSION=1.0.1 conda update -c numba numba -y &&
  pip install --upgrade PyYAML
06/10/2021 12:56 AM UTC
4bbecd02b1be34c28dde9347c0ae8ccbae2319c13dcffefb012dec3cdeb02137RUN
NEMO_VERSION=1.0.1 cd /tmp/nemo &&
  pip install ".[all]" &&
  python -c "import nemo.collections.asr as nemo_asr" &&
  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"
06/10/2021 12:55 AM UTC
73af065613d798fc79be4cc6958faae00122ae5c90845011413097d762c68100RUN
NEMO_VERSION=1.0.1 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
06/10/2021 12:55 AM UTC
b8f3229587207c1f03b950b579577ba857a04ab16254a3a1dc473040bd9359deARG
NEMO_VERSION=1.0.1
06/10/2021 12:55 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.