From e138ffff18ca06c7119f8e8deb2d2a2df7e876e7 Mon Sep 17 00:00:00 2001 From: RaulPPealez Date: Wed, 28 Jun 2023 15:36:34 +0200 Subject: [PATCH] Try setting archs to OFF --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 08bd194..42f5cc0 100644 --- a/setup.py +++ b/setup.py @@ -155,7 +155,7 @@ def build_extension(self, ext: CMakeExtension) -> None: else: raise RuntimeError("Unsupported CUDA version") CMAKE_CUDA_ARCHS = ";".join([str(arch) for arch in ARCHES] + [f"{LATEST_ARCH}-real", f"{LATEST_ARCH}-virtual"]) - extra_args["CMAKE_CUDA_ARCHITECTURES"] = CMAKE_CUDA_ARCHS + extra_args["CMAKE_CUDA_ARCHITECTURES"] = "OFF" #CMAKE_CUDA_ARCHS else: extra_args["ENABLE_CUDA"] = "OFF"