torch-cuda-arch-list (#22)

predibase · Nov 16, 2023 · f76119a · f76119a
1 parent 8c75da0
commit f76119a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -150,6 +150,7 @@ WORKDIR /usr/src
 
 COPY server/Makefile-punica Makefile
 
+ENV TORCH_CUDA_ARCH_LIST="8.0;8.6+PTX"
 # Build specific version of punica
 RUN make build-punica