mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
27 lines
874 B
Python
27 lines
874 B
Python
## @package get_python_cmake_flags
|
|
# Module scripts.get_python_cmake_flags
|
|
##############################################################################
|
|
# Use this script to find your preferred python installation.
|
|
##############################################################################
|
|
#
|
|
# You can use the following to build with your preferred version of python
|
|
# if your installation is not being properly detected by CMake.
|
|
#
|
|
# mkdir -p build && cd build
|
|
# cmake $(python ../scripts/get_python_cmake_flags.py) ..
|
|
# make
|
|
#
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
from __future__ import print_function
|
|
from distutils import sysconfig
|
|
import sys
|
|
|
|
flags = [
|
|
'-DPYTHON_EXECUTABLE:FILEPATH={}'.format(sys.executable),
|
|
'-DPYTHON_INCLUDE_DIR={}'.format(sysconfig.get_python_inc()),
|
|
]
|
|
|
|
print(' '.join(flags), end='')
|