Try setting archs to OFF

openmm · Jun 28, 2023 · e138fff · e138fff
1 parent 4a4b475
commit e138fff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"