update

pyg-team · Mar 24, 2024 · 5043350 · 5043350
1 parent c68b7b8
commit 5043350
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -59,7 +59,8 @@ def build_extension(self, ext):
             '-DBUILD_TEST=OFF',
             '-DBUILD_BENCHMARK=OFF',
             '-DUSE_PYTHON=ON',
-            f'-DPython3_ROOT_DIR={sys.exec_prefix}',
+            f'-DPython_ROOT_DIR={sys.exec_prefix}',
+            f'-DPython_EXECUTABLE={sys.executable}',
             f'-DWITH_CUDA={"ON" if WITH_CUDA else "OFF"}',
             f'-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={extdir}',
             f'-DCMAKE_BUILD_TYPE={self.build_type}',