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
b81ca4a9458259119355c9bfd26b35f1bbc1234b9d4169506460040aba067e19CONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
10/05/2020 5:06 PM UTC
41247b28263cb6147566d4ac368b8db40cffaefb5653f51438c7cc3bef338c06COPY
tutorials /workspace/nemo/tutorials
10/05/2020 5:06 PM UTC
f7bd57afef1ca523224590a4ba1c62c72a673d23b20954ec63b8a8353ff7c2a0COPY
tests /workspace/nemo/tests
10/05/2020 5:06 PM UTC
6aaaa47954f2f96b9543fd7f956d13660feec942d5fbc87302ef180e81385acbCOPY
examples /workspace/nemo/examples
10/05/2020 5:06 PM UTC
2d5045b383168c2af24ec6a8184d248ef4397bdb66f509791cdfc5f45aa38ba5COPY
scripts /workspace/nemo/scripts
10/05/2020 5:06 PM UTC
48c91c2923e2b16b843fdb7e87f248dd8875b428906fd37371650509a17a4f8bWORKDIR
/workspace/nemo
10/05/2020 5:06 PM UTC
d7b3153ce22d7f9c62dd663cf6c2ec66ae2795387c2ed63c096dfea6e88eab99RUN
NEMO_VERSION=1.0.0b1 cd /tmp/nemo &&
  pip install ".[all]"
10/05/2020 5:06 PM UTC
ce88d4d734896c7763e02bcfd77278919d9ebf35d1ce4eedb624d1acb658487eRUN
NEMO_VERSION=1.0.0b1 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
10/05/2020 4:59 PM UTC
ba8ad1478822306c1968a8a4cf298784180471800365621821399d9ce126ebf2ARG
NEMO_VERSION=1.0.0b1
10/05/2020 4:59 PM UTC
e8df182f1a7fca94116df6fcc5110bcc40dd5fdfea8e4f14c47592d25ff698e8RUN
for f in $(ls requirements/*.txt); do pip install --disable-pip-version-check --no-cache-dir -r $f; done
10/05/2020 4:59 PM UTC
...