From 8b49d0cf067104d190b1cbac653457794bc60d8f Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Mon, 30 Sep 2024 15:02:21 -0700 Subject: [PATCH] update workdir --- Dockerfile | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 80f2c82b10..bf3d2015a1 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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"