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
58a5c348a3d2c4c151af5724bde86977597fa2a6a14ce01520dc591706505f46CONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
04/06/2021 9:30 PM UTC
86523781600a16b7b5ea3cf6f312ca3398b4697a310b838cbd8827841f3c4977COPY
tutorials /workspace/nemo/tutorials
04/06/2021 9:30 PM UTC
365ead8ea79c49df701495ede321c9f010aa97570969b8544ec505776903d4b8COPY
tests /workspace/nemo/tests
04/06/2021 9:30 PM UTC
7ebb0027070a09b2b0d0acd466bb635d9b614d5c9d2020e12daa0bc167835674COPY
examples /workspace/nemo/examples
04/06/2021 9:30 PM UTC
5bb910c772bd1f6b5d600b35dbb99eea45186b437370017cb475fe339bdfab25COPY
scripts /workspace/nemo/scripts
04/06/2021 9:30 PM UTC
caad2ff19c71d36a0a6a642b692f6400d8f1eda6d2eb3f1e3e759599a0a24c0eWORKDIR
/workspace/nemo
04/06/2021 9:30 PM UTC
300374f55950ee626fcfc7e1418b8c04d7139361ffa66bbca56fc685e63663e6RUN
NEMO_VERSION=1.0.0rc1 cd /tmp/nemo &&
  pip install ".[all]"
04/06/2021 9:30 PM UTC
21e9dd7da481c7ff5e40e4fc62188e73c8deed324165e3987233ffd719dcf43dRUN
NEMO_VERSION=1.0.0rc1 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
04/06/2021 9:28 PM UTC
d7291ac20a6649a50bf4ca31ee812b743263f00fc81e254ec34f346628e8b043ARG
NEMO_VERSION=1.0.0rc1
04/06/2021 9:28 PM UTC
9f52a112694858644f122e798232a42fa5632843c073b35e128683b65279a1f1RUN
curl -Lo- https://raw.githubusercontent.com/nvm-sh/nvm/v0.37.2/install.sh | bash &&
  source "$NVM_DIR/nvm.sh" &&
  nvm install 15.12.0 node &&
  nvm use 15.12.0 &&
  nvm ls &&
  nvm alias default 15.12.0 &&
  nvm uninstall 15.0.1 &&
  npm update ssri &&
  npm update y18n
04/06/2021 9:28 PM UTC
...