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
d4294c6d6245dc212b9f4e3482d9566975be0a48e800bd692eab2359f2b70ee7CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
01/04/2023 7:14 PM UTC
308ae7fec03e890f1e1ba6fefd0d2339d53c8f86f588f1e032ab9aab12357691RUN
NEMO_VERSION=1.14.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
01/04/2023 7:14 PM UTC
b3c49383cf583a93fb6ffd1627eca47b283e9af831b6a29b0abed617c66c82e4COPY
tutorials /workspace/nemo/tutorials
01/04/2023 7:14 PM UTC
18c7cef6866de7171b031435f61a4b61f4f13ee8fbb2ef91efe0924eac10c202COPY
tests /workspace/nemo/tests
01/04/2023 7:14 PM UTC
ff41dbf04f61f00894d04be5b5407e995fb4e9337926aa28974d5edff4ad2785COPY
examples /workspace/nemo/examples
01/04/2023 7:14 PM UTC
201f41d61d9dd00bf1f66b553a654506a2e8ab0c2acb2e326893ef7605c95b7cCOPY
scripts /workspace/nemo/scripts
01/04/2023 7:14 PM UTC
fd67a6ef1150d3fa0abf903722eb052d5b670e70af1334acb9146e0684d60b7bWORKDIR
/workspace/nemo
01/04/2023 7:14 PM UTC
e73daf66a8dffc34a9702d8c417937b649ebd56b99f88058eea7aa6aa3291755RUN
NEMO_VERSION=1.14.0 pip install numba==0.53.1
01/04/2023 7:14 PM UTC
2d0ea671574e6ace6e04835a50ab229c3866de82d5c3f1f6cbb7b4dc9eb09df3RUN
NEMO_VERSION=1.14.0 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"
01/04/2023 7:14 PM UTC
68ffa5433da7664e66139f72c082ea3a75a2f857c84a4c0ec0827bee283c2b2dRUN
NEMO_VERSION=1.14.0 cd /tmp/nemo &&
  pip install ".[all]"
01/04/2023 7:14 PM UTC
...