diff --git a/Dockerfile-xformers b/Dockerfile-xformers
index 1d1a07d..2e497c5 100644
--- a/Dockerfile-xformers
+++ b/Dockerfile-xformers
@@ -31,7 +31,7 @@ WORKDIR /root
 RUN git clone --depth=1 --no-tags --recurse-submodules --shallow-submodules \
         https://github.com/facebookresearch/xformers.git 
 
-ENV TORCH_CUDA_ARCH_LIST="6.1;7.5;8.0;8.6"
+ARG TORCH_CUDA_ARCH_LIST="6.1;7.5;8.0;8.6"
 
 WORKDIR /root/xformers
 RUN --mount=type=cache,target=/root/.cache/pip \