'xformers' get TORCH_CUDA_ARCH_LIST properly

YanWenKun · Mar 23, 2023 · 4d020a2 · 4d020a2
1 parent 517310b
commit 4d020a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 \