f

Tencent · Dec 20, 2024 · 7146287 · 7146287
1 parent f521563
commit 7146287
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tools/pnnx/CMakeLists.txt b/tools/pnnx/CMakeLists.txt
@@ -38,6 +38,8 @@ option(PNNX_COVERAGE "build for coverage" OFF)
 # set(Torch_DIR "${Torch_INSTALL_DIR}/share/cmake/Torch")
 set(TorchVision_DIR "${TorchVision_INSTALL_DIR}/share/cmake/TorchVision")
 
+set(Python_FIND_REGISTRY "LAST")
+set(Python_FIND_FRAMEWORK "LAST")
 find_package(Python3 COMPONENTS Interpreter Development)
 
 PNNXProbeForPyTorchInstall()

diff --git a/tools/pnnx/cmake/PNNXPyTorch.cmake b/tools/pnnx/cmake/PNNXPyTorch.cmake
@@ -11,8 +11,6 @@ function(PNNXProbeForPyTorchInstall)
     message(STATUS "Using cached Torch install dir = ${Torch_INSTALL_DIR}")
     set(Torch_DIR "${Torch_INSTALL_DIR}/share/cmake/Torch" CACHE STRING "Torch dir" FORCE)
   else()
-    set(Python_FIND_REGISTRY "LAST")
-    set(Python_FIND_FRAMEWORK "LAST")
     #find_package (Python3 COMPONENTS Interpreter Development)
     find_package (Python3)
     message(STATUS "Checking for PyTorch using ${Python3_EXECUTABLE} ...")