From bf032320f4e23e20731ed817b9a113db891b1cc4 Mon Sep 17 00:00:00 2001
From: littlemine <wxlwxl1993@zju.edu.cn>
Date: Tue, 9 Jul 2024 15:54:07 +0800
Subject: [PATCH] setup numpy during build

---
 projects/CUDA/CMakeLists.txt | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/projects/CUDA/CMakeLists.txt b/projects/CUDA/CMakeLists.txt
index 29f8d9f5da..3acd821246 100644
--- a/projects/CUDA/CMakeLists.txt
+++ b/projects/CUDA/CMakeLists.txt
@@ -48,16 +48,21 @@ if (ZS_PYTHON_FOUND AND ZENO_WITH_PyZpc)
   message(STATUS "python3 libraries: ${ZS_OVERWRITE_PYTHON_LIBRARIES}")
   message(STATUS "python3 include directories: ${ZS_OVERWRITE_PYTHON_INCLUDE_DIR}")
   set(ENV{ZENO_BIN_DIR} "${ZENO_BIN_DIR}")
-  #[=[
   add_custom_command(
     TARGET copy_py
     POST_BUILD
     COMMAND 
     ${CMAKE_COMMAND} -E env ZENO_BIN_DIR="${ZENO_BIN_DIR}"
-    ${ZS_OVERWRITE_PYTHON_EXECUTABLE} -m pip install ${CMAKE_CURRENT_SOURCE_DIR}/zpc_jit --user
-    COMMENT "installing pyzpc into the specified conda environment: ${ZS_OVERWRITE_PYTHON_EXECUTABLE}"
+    ${ZS_OVERWRITE_PYTHON_EXECUTABLE} -m ensurepip
+    COMMENT "preparing pip module"
+  )
+  add_custom_command(
+    TARGET copy_py
+    POST_BUILD
+    COMMAND 
+    ${ZS_OVERWRITE_PYTHON_EXECUTABLE} -m pip install numpy --user
+    COMMENT "installing numpy"
   )
-  ]=]
   set(PY_LIBS_DIR ${RESOURCE_BASE_DIR}/resource/py_libs)
   set(PREFIX "lib")
   if (WIN32)