Fix pythonlib can not be found

purplegourd · Nov 25, 2016 · aa6c6dc · aa6c6dc
1 parent 5e7e5cc
commit aa6c6dc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -12,8 +12,12 @@ include(package)
 find_package(SWIG 2.0)
 find_package(CUDA QUIET)
 find_package(Protobuf REQUIRED)
-find_package(PythonLibs 2.7 REQUIRED)
+
+# Set up the versions we know about, in the order we will search.
+# Always add the user supplied additional versions to the front.
+set(Python_ADDITIONAL_VERSIONS 2.7)
 find_package(PythonInterp 2.7 REQUIRED)
+find_package(PythonLibs 2.7 REQUIRED)
 find_package(ZLIB REQUIRED)
 find_package(NumPy REQUIRED)
 find_package(Threads REQUIRED)