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
cf6ed3e2528247e51046ee400d654152634bc62614f62b411b862acc8c7c7f17CONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
11/16/2020 10:42 PM UTC
c8cff2dc3bddb3c11901f84d4d2c2ff508e68f74b6ff39cb2314b87f3b6bc10fCOPY
tutorials /workspace/nemo/tutorials
11/16/2020 10:42 PM UTC
bdf299c8409beb19bb59e4d14f138ea039d1be486c0ede782db9a5582f206029COPY
tests /workspace/nemo/tests
11/16/2020 10:42 PM UTC
dbb77fb8f05b663f0b211311df237c4f20c025ac0d57844f23c3d407fbd2f8f2COPY
examples /workspace/nemo/examples
11/16/2020 10:42 PM UTC
d12d64971d24d787e086d21b72c207f040df4a49259ee2b6359e57cf6b5f45ffCOPY
scripts /workspace/nemo/scripts
11/16/2020 10:42 PM UTC
ad9a688b2725f1ff9500c39e231280b5919b5e93d5231783ebfffbfc131e71adWORKDIR
/workspace/nemo
11/16/2020 10:42 PM UTC
96277d7bafd1f0a16ec7e48e19f80d7de830c274d4fa510a247392f3c20f2b5cRUN
NEMO_VERSION=stable-76-g624099bd cd /tmp/nemo &&
  pip install ".[all]"
11/16/2020 10:42 PM UTC
d09865793ad0afa4e8d37379609f9ea819498f067b669e8e1c378f7262d44fc9RUN
NEMO_VERSION=stable-76-g624099bd /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
11/16/2020 10:42 PM UTC
e2581f5e87e040b73c25beac6d4ff7072e6131d85ec7f25f27d93c82cd280251ARG
NEMO_VERSION=1.0.0b2
11/16/2020 10:42 PM UTC
0374e6177b249a5394dfba0106cc5955fba6914f58e1f0c09dae1c5dae8323a6RUN
./install_ctc_decoders.sh
11/16/2020 10:42 PM UTC
...