1717ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.07-py3
1818
1919FROM ${BASE_IMAGE} AS base-image
20- ENV PIP_CONSTRAINT=""
20+ ENV PIP_CONSTRAINT=""
2121ARG IMAGE_LABEL
2222LABEL "nemo.library"=${IMAGE_LABEL}
2323
@@ -26,7 +26,7 @@ ENV HYDRA_FULL_ERROR=1
2626ENV PYTHONUNBUFFERED=1
2727
2828# APT packages
29- RUN <<"EOF" bash -ex
29+ RUN bash -ex <<"EOF"
3030apt-get update
3131apt-get install -y bc
3232apt-get clean
3535WORKDIR /tmp/NeMo
3636ARG TRTLLM_REPO
3737ARG TRTLLM_TAG
38- RUN --mount=type=bind,source=docker/common/install_dep.sh,target=/tmp/NeMo/install_dep.sh <<"EOF" bash -ex
38+ RUN --mount=type=bind,source=docker/common/install_dep.sh,target=/tmp/NeMo/install_dep.sh bash -ex <<"EOF"
3939
4040 bash /tmp/NeMo/install_dep.sh --library trt --mode install
4141EOF
4242
43-
4443FROM base-image AS trt-llm-wheel
4544WORKDIR /tmp/NeMo
4645ARG TRTLLM_REPO
4746ARG TRTLLM_TAG
48- RUN --mount=type=bind,source=docker/common/install_dep.sh,target=/tmp/NeMo/install_dep.sh <<"EOF" bash -ex
47+ RUN --mount=type=bind,source=docker/common/install_dep.sh,target=/tmp/NeMo/install_dep.sh bash -ex <<"EOF"
4948
5049 bash /tmp/NeMo/install_dep.sh --library trtllm --mode build
5150EOF
@@ -55,7 +54,7 @@ WORKDIR /tmp/NeMo
5554ARG TE_REPO
5655ARG TE_TAG
5756RUN --mount=type=bind,source=docker/common/install_dep.sh,target=/tmp/NeMo/install_dep.sh \
58- --mount=type=bind,source=external/patches,target=/tmp/NeMo/external/patches <<"EOF" bash -ex
57+ --mount=type=bind,source=external/patches,target=/tmp/NeMo/external/patches bash -ex <<"EOF"
5958
6059 bash /tmp/NeMo/install_dep.sh --library te --mode build
6160 ls -al /tmp/Megatron-LM || true
@@ -65,7 +64,7 @@ FROM base-image as mcore-wheel
6564WORKDIR /tmp/NeMo
6665ARG MLM_REPO
6766ARG MLM_TAG
68- RUN --mount=type=bind,source=docker/common/install_dep.sh,target=/tmp/NeMo/install_dep.sh <<"EOF" bash -ex
67+ RUN --mount=type=bind,source=docker/common/install_dep.sh,target=/tmp/NeMo/install_dep.sh bash -ex <<"EOF"
6968
7069 bash /tmp/NeMo/install_dep.sh --library mcore --mode build
7170 ls -al /tmp/Megatron-LM || true
@@ -85,14 +84,14 @@ RUN \
8584 --mount=type=bind,source=external/patches,target=/tmp/NeMo/external/patches \
8685 --mount=type=bind,source=README.md,target=/tmp/NeMo/README.md \
8786 --mount=type=bind,source=nemo/package_info.py,target=/tmp/NeMo/nemo/package_info.py \
88- --mount=type=bind,source=nemo/__init__.py,target=/tmp/NeMo/nemo/__init__.py <<"EOF" bash -ex
87+ --mount=type=bind,source=nemo/__init__.py,target=/tmp/NeMo/nemo/__init__.py bash -ex <<"EOF"
8988
9089 bash /tmp/NeMo/install_dep.sh --library all --mode install
9190 pip install --no-cache-dir ".[deploy,test]"
9291 rm -rf $NEMO_DIR || true
92+
9393EOF
9494
9595WORKDIR /workspace
9696ENV PYTHONPATH="${PYTHONPATH}:/workspace/Megatron-LM"
9797ENV NEMO_HOME="/home/TestData/nemo_home"
98-
0 commit comments