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"