fix a cmakelist bug, find python first then find TF

deepmodeling · Dec 14, 2022 · 15541f9 · 15541f9
1 parent ada465e
commit 15541f9
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/source/CMakeLists.txt b/source/CMakeLists.txt
@@ -77,6 +77,11 @@ else()
   message(STATUS "Will not build AMD GPU support")  
 endif (USE_ROCM_TOOLKIT)
 
+# setup tensorflow libraries by python
+if (USE_TF_PYTHON_LIBS)
+  find_package (Python COMPONENTS Interpreter Development REQUIRED)
+endif(USE_TF_PYTHON_LIBS)
+
 #define USE_ASCEND, find tensorflow abi info
 if (USE_ASCEND)
   add_definitions("-DHUAWEI_ASCEND")
@@ -123,11 +128,6 @@ endif(USE_ASCEND AND BUILD_CPP_IF)
 
 set(DEEPMD_SOURCE_DIR ${PROJECT_SOURCE_DIR}/..)
 
-# setup tensorflow libraries by python
-if (USE_TF_PYTHON_LIBS)
-  find_package (Python COMPONENTS Interpreter Development REQUIRED)
-endif(USE_TF_PYTHON_LIBS)
-
 if (TENSORFLOW_VERSION VERSION_GREATER_EQUAL 2.10)
   set (CMAKE_CXX_STANDARD 17)
 elseif (TENSORFLOW_VERSION VERSION_GREATER_EQUAL 2.7)