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
3f109b486b6b3bc7b18245ec254d9c71e1d9315e9047d3efb78ea4bc59574cb5CONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
07/09/2020 11:15 PM UTC
9da1606c774a0474b4b98584d3008fe148ea6f79c37e56aa9ef63c1ccd999231COPY
README.rst LICENSE /workspace/nemo/
07/09/2020 11:15 PM UTC
af0bdc7539d6ac58fe68044c10dfd294e493c3167ff7dc29f81ff94e694c5806COPY
tests /workspace/nemo/tests
07/09/2020 11:15 PM UTC
79cb346e3430cf23fa2ba1007e75612b65a6c4a5277e6002795f03f530d107feCOPY
examples /workspace/nemo/examples
07/09/2020 11:15 PM UTC
711a74bb1d79a5ae8888bdb26d0a42e46f2d9f1c1522cb029e9783a607ff1396COPY
scripts /workspace/nemo/scripts
07/09/2020 11:15 PM UTC
5acda49d63ebde477db9020c3901dfd772a9078541c51a9dc6a66dbd094f0c0dWORKDIR
/workspace/nemo
07/09/2020 11:15 PM UTC
ad5bfa78a5b1c980dc792301ac0ad53ad34472d6ea22aaae8038b2e2c0c5b5e7RUN
NEMO_VERSION=v0.10.1-428-g4b7bbc4a BASE_IMAGE=nvcr.io/nvidia/pytorch:20.01-py3 cd /tmp/nemo &&
  pip install ".[all]"
07/09/2020 11:15 PM UTC
268dea3537feef2f4b55df378ad4f19fb98dc4adfd0d9f910ac77d3720272e18RUN
NEMO_VERSION=v0.10.1-428-g4b7bbc4a BASE_IMAGE=nvcr.io/nvidia/pytorch:20.01-py3 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
07/09/2020 11:15 PM UTC
88099e511d7dd05fe315ffe38b185304f2b7616f569d8ee55b1928cd4a61303eARG
BASE_IMAGE
07/01/2020 10:21 PM UTC
caacf472e7e685c6bd7be865a6da539ca5eb2642a484e6d1241aef9d6360eadeARG
NEMO_VERSION
07/01/2020 10:21 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.