Skip to content

Commit

Permalink
workdir
Browse files Browse the repository at this point in the history
  • Loading branch information
v-chen_data committed Sep 30, 2024
1 parent 8b49d0c commit 81296f3
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,6 @@ ARG DEP_GROUPS
ARG TE_COMMIT
ARG KEEP_FOUNDRY=false

WORKDIR /

ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 8.9 9.0"

# Check for changes in setup.py.
Expand All @@ -22,8 +20,9 @@ RUN rm setup.py
RUN NVTE_FRAMEWORK=pytorch CMAKE_BUILD_PARALLEL_LEVEL=4 MAX_JOBS=4 pip install git+https://github.com/NVIDIA/TransformerEngine.git@$TE_COMMIT

# Install and uninstall foundry to cache foundry requirements
WORKDIR /
RUN git clone -b $BRANCH_NAME https://github.com/mosaicml/llm-foundry.git
RUN pip install --no-cache-dir "./llm-foundry${DEP_GROUPS}"
RUN pip install --no-cache-dir "/llm-foundry${DEP_GROUPS}"

# Conditionally uninstall llm-foundry and remove its directory
RUN if [ "$KEEP_FOUNDRY" != "true" ]; then \
Expand Down

0 comments on commit 81296f3

Please sign in to comment.