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
a8209a25b31fb7a5780139401a8d19aab95b9b81fa2afeeae5736bd68e06b7eaCONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
03/24/2020 5:40 PM UTC
8a273790f2d60f0d45dff53554927f16641611909ce20e4ad03de00474696455COPY
README.rst LICENSE /workspace/nemo/
03/24/2020 5:40 PM UTC
5bb49e94b8ff4d6614a46d0d30f01f1b217a8d5b30e510a98bf5827ea93e9fcfCOPY
tests /workspace/nemo/tests
03/24/2020 5:40 PM UTC
df3cf271c627ef3d65a870befd61215d0f1bc66dd450ec307350c8fe7660c4f0COPY
examples /workspace/nemo/examples
03/24/2020 5:40 PM UTC
f96fcc05cd26f9a04b9c8d39f1b29ecf30dfe161d4d36d4d9af3c4be4b7fe85cCOPY
scripts /workspace/nemo/scripts
03/24/2020 5:40 PM UTC
bd807974933e5f28770f8714fb2d99e37ca005325ef6f2931e27aa2695545afbWORKDIR
/workspace/nemo
03/24/2020 5:40 PM UTC
22105a78c740fdf3e2941fa763af107bfea60c87eed3da8992a3b48bc8cde4e0RUN
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
c466672225787fe54950927d997d244b38821b3baff5aefa23160e100b5e6485RUN
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
b80cdbfe2a31194c29c165109038fe7d79dfff7b3c326ee550072d74c85322abCOPY
/tmp/trt-oss/TensorRT/build/libnvinfer_plugin.so* /usr/lib/x86_64-linux-gnu/
03/24/2020 5:40 PM UTC
e495c8e2ae9062c359551c2202144a7fe16cbb8942ef7ccfae188dad3cd6c84eARG
BASE_IMAGE
03/24/2020 5:40 PM UTC
...