Caffe2 - Python API
A deep learning, cross platform ML framework
get_python_cmake_flags.py
1 
14 
15 from __future__ import absolute_import
16 from __future__ import unicode_literals
17 from __future__ import print_function
18 from distutils import sysconfig
19 import os
20 import sys
21 import platform
22 
23 version = platform.python_version()
24 if version[:3] != '2.7':
25  print('ERROR: Python {version} is not officially supported yet.'
26  .format(version=version), file=sys.stderr)
27  exit(1)
28 
29 # Flags to print to stdout
30 flags = ''
31 inc = sysconfig.get_python_inc()
32 lib = sysconfig.get_config_var("LIBDIR")
33 
34 # macOS specific
35 if sys.platform == "darwin":
36  lib = os.path.dirname(lib) + '/Python'
37  if os.path.isfile(lib):
38  flags += '-DPYTHON_LIBRARY={lib}'.format(lib=lib)
39 
40 if os.path.isfile(inc + '/Python.h'):
41  flags += '-DPYTHON_INCLUDE_DIR={inc}'.format(inc=inc)