diff options
Diffstat (limited to 'tensorflow')
| -rw-r--r-- | tensorflow/contrib/cmake/CMakeLists.txt | 6 | ||||
| -rwxr-xr-x | tensorflow/contrib/cmake/tf_python.cmake | 1 |
2 files changed, 6 insertions, 1 deletions
diff --git a/tensorflow/contrib/cmake/CMakeLists.txt b/tensorflow/contrib/cmake/CMakeLists.txt index af7b4fb386..83b405fd4e 100644 --- a/tensorflow/contrib/cmake/CMakeLists.txt +++ b/tensorflow/contrib/cmake/CMakeLists.txt @@ -229,6 +229,12 @@ if (tensorflow_ENABLE_GPU) endif() endif() +# Find python executable +include(FindPythonInterp) +if(NOT ${PYTHONINTERP_FOUND}) + message(FATAL_ERROR "CMake was unable to find a python interpreter.") +endif() + # Let's get to work! include(tf_core_framework.cmake) # NOTE: Disabled until issue #3996 is fixed. diff --git a/tensorflow/contrib/cmake/tf_python.cmake b/tensorflow/contrib/cmake/tf_python.cmake index 9e2eb71b4c..bfd6a18c78 100755 --- a/tensorflow/contrib/cmake/tf_python.cmake +++ b/tensorflow/contrib/cmake/tf_python.cmake @@ -27,7 +27,6 @@ # 1. Resolve the installed version of Python (for Python.h and python). # TODO(mrry): Parameterize the build script to enable Python 3 building. -include(FindPythonInterp) if(NOT PYTHON_INCLUDE_DIR) set(PYTHON_NOT_FOUND false) exec_program("${PYTHON_EXECUTABLE}" |