find PYTHON_VERSION first

pytorch · Mar 5, 2021 · 0b76f5f · 0b76f5f
1 parent 07ca402
commit 0b76f5f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchaudio/csrc/CMakeLists.txt b/torchaudio/csrc/CMakeLists.txt
@@ -103,6 +103,7 @@ if (BUILD_TORCHAUDIO_PYTHON_EXTENSION)
   find_library(TORCH_PYTHON_LIBRARY torch_python PATHS "${TORCH_INSTALL_PREFIX}/lib")
 
   if (WIN32)
+    set(Python_ADDITIONAL_VERSIONS ${PYTHON_VERSION} 3.9 3.8 3.7 3.6)
     find_package(Python3 COMPONENTS Development)
     set(ADDITIONAL_ITEMS Python3::Python)
   endif()