Update CMakeLists.txt

media2x · Mar 7, 2022 · 3a33e0d · 3a33e0d
1 parent c05d2e7
commit 3a33e0d
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/rife_ncnn_vulkan_python/CMakeLists.txt b/rife_ncnn_vulkan_python/CMakeLists.txt
@@ -16,10 +16,18 @@ find_package(OpenMP)
 find_package(Vulkan REQUIRED)
 
 # Python
-if(${CMAKE_VERSION} VERSION_LESS "3.15")
-  find_package(Python REQUIRED)
+if (DEFINED PY_VERSION)
+  if (${CMAKE_VERSION} VERSION_LESS "3.15")
+  	find_package(Python ${PY_VERSION} EXACT REQUIRED)
+  else()
+   	find_package(Python ${PY_VERSION} EXACT REQUIRED COMPONENTS Development)
+  endif()
 else()
-  find_package(Python REQUIRED COMPONENTS Development)
+  if (${CMAKE_VERSION} VERSION_LESS "3.15")
+    find_package(Python REQUIRED)
+  else()
+    find_package(Python REQUIRED COMPONENTS Development)
+  endif()
 endif()
 
 # SWIG