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
sha256:b499944b48333ece92b5ed4a5ee5f4c245c11511057068d307dc1c6bf19c76a1RUN
NEMO_VERSION=v0.9.0-1149-g4884cb37 BASE_IMAGE=nvcr.io/nvidia/pytorch:20.01-py3 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
03/24/2020 5:40 PM UTC
sha256:e6c7893b783323fc91f6fc1df6893d432ac260387abd51363cc1ce7e63deb2c2COPY
README.rst LICENSE /workspace/nemo/
03/24/2020 5:40 PM UTC
sha256:3868510fa7c8d126c1c22743bf4a280b7e5d57ebefb38c5139c76251871063beCOPY
tests /workspace/nemo/tests
03/24/2020 5:40 PM UTC
sha256:6ec46c65f56aaba5a50de2c4530747002f89eb841055eafefba385c396cf597eCOPY
examples /workspace/nemo/examples
03/24/2020 5:40 PM UTC
sha256:76312294607792b4416269a3ec62da7a8d284196a79c4d97119490083b72a461COPY
scripts /workspace/nemo/scripts
03/24/2020 5:40 PM UTC
sha256:12b0fe7d785dd0142f0570f9ae32982c9a6158e8d80b7d447cfcc35e35ee0547WORKDIR
/workspace/nemo
03/24/2020 5:40 PM UTC
sha256:44f75618f7f2bb5c1c228ac13cda3f6224e1cfb675652ccb8b6564ae0146eb2aRUN
NEMO_VERSION=v0.9.0-1149-g4884cb37 BASE_IMAGE=nvcr.io/nvidia/pytorch:20.01-py3 cd /tmp/nemo &&
  pip install ".[all]"
03/24/2020 5:40 PM UTC
sha256:3f6d4a514d3c82b152f56797f9f302d957c5e570d8238ad67fad7ff85b4f0a3bRUN
NEMO_VERSION=v0.9.0-1149-g4884cb37 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
03/24/2020 5:40 PM UTC
sha256:b9de7368b9679a30f636b3c48bc379ca996d237140cf1e09b116fea492a7f7dfCOPY
/tmp/trt-oss/TensorRT/build/libnvinfer_plugin.so* /usr/lib/x86_64-linux-gnu/
03/24/2020 5:40 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
BASE_IMAGE
03/24/2020 5:40 PM UTC
...