Skip to content

Commit

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

RUN echo "KEEP_FOUNDRY is set to: '$KEEP_FOUNDRY'"
ENV KEEP_FOUNDRY=${KEEP_FOUNDRY}
WORKDIR /

ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 8.9 9.0"

Expand Down

0 comments on commit 8b49d0c

Please sign in to comment.