[Cmake] Auto detect python in env. (#43)

Signed-off-by: Duyi-Wang <[email protected]>

Author: Duyi-Wang

src/CMakeLists.txt

@@ -37,7 +37,10 @@ target_link_libraries(xfastertransformer
 add_dependencies(xfastertransformer_static ${DEPEND_LIST})
 add_dependencies(xfastertransformer ${DEPEND_LIST})
 
-execute_process(COMMAND python -m pip show torch
+find_package (Python COMPONENTS Interpreter Development)
+message(STATUS "Using python: ${Python_EXECUTABLE}")
+
+execute_process(COMMAND ${Python_EXECUTABLE} -m pip show torch
                 RESULT_VARIABLE EXIT_CODE
                 OUTPUT_QUIET)
 if(${EXIT_CODE} EQUAL 0)

src/pytorch/CMakeLists.txt

@@ -18,7 +18,7 @@ find_package(OpenMP REQUIRED)
 find_package(MPI REQUIRED)
 find_package(oneCCL REQUIRED)
 
-execute_process(COMMAND python -c "import torch.utils; print(torch.utils.cmake_prefix_path)"
+execute_process(COMMAND ${Python_EXECUTABLE} -c "import torch.utils; print(torch.utils.cmake_prefix_path)"
                 OUTPUT_VARIABLE TORCH_CMAKE_PREFIX_PATH
                 OUTPUT_STRIP_TRAILING_WHITESPACE)