pytorch

1
## @package get_python_cmake_flags
2
# Module scripts.get_python_cmake_flags
3
##############################################################################
4
# Use this script to find your preferred python installation.
5
##############################################################################
6
#
7
# You can use the following to build with your preferred version of python
8
# if your installation is not being properly detected by CMake.
9
#
10
#   mkdir -p build && cd build
11
#   cmake $(python ../scripts/get_python_cmake_flags.py) ..
12
#   make
13
#
14

15

16
import sys
17
import sysconfig
18

19
flags = [
20
    f"-DPYTHON_EXECUTABLE:FILEPATH={sys.executable}",
21
    f"-DPYTHON_INCLUDE_DIR={sysconfig.get_path('include')}",
22
]
23

24
print(" ".join(flags), end="")
25