diff --git a/setup.py b/setup.py
index b369d3c..749f12a 100644
--- a/setup.py
+++ b/setup.py
@@ -153,7 +153,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"] = "50"
+    extra_args["CMAKE_CUDA_ARCHITECTURES"] = CMAKE_CUDA_ARCHS
 else:
     extra_args["ENABLE_CUDA"] = "OFF"