diff --git a/setup.py b/setup.py
index c72553a4650c7671669fa144ab3baacad77373fd..cb6137da4d29dcfc1ecf5a78481ecf984ea8e71e 100644
--- a/setup.py
+++ b/setup.py
@@ -87,6 +87,7 @@ class CMakeBuild(build_ext):
                 f"-DENABLE_ASAN={asan}",
                 f"-DCUDA={with_cuda}",
                 "-DPYBIND=ON",
+                "-DPYBIND11_FINDPYTHON=ON",
                 "-DCMAKE_EXPORT_COMPILE_COMMANDS=ON",
                 "-DCOVERAGE=OFF",
                 "-DCMAKE_CUDA_ARCHITECTURES=native",