Заставьте distutils искать numpy заголовочные файлы в корректном месте

В моей установке, numpy's arrayobject.h расположен в …/site-packages/numpy/core/include/numpy/arrayobject.h. Я записал тривиальный сценарий Cython, который использует numpy:

cimport numpy as np

def say_hello_to(name):
    print("Hello %s!" % name)

У меня также есть следующий distutils setup.py (скопированный с руководства пользователя Cython):

from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext

ext_modules = [Extension("hello", ["hello.pyx"])]

setup(
  name = 'Hello world app',
  cmdclass = {'build_ext': build_ext},
  ext_modules = ext_modules
)

Когда я пытаюсь создать с python setup.py build_ext --inplace, Cython пытается сделать следующее:

gcc -fno-strict-aliasing -Wno-long-double -no-cpp-precomp -mno-fused-madd \
-fno-common -dynamic -DNDEBUG -g -Os -Wall -Wstrict-prototypes -DMACOSX \
-I/usr/include/ffi -DENABLE_DTRACE -arch i386 -arch ppc -pipe \
-I/System/Library/Frameworks/Python.framework/Versions/2.5/include/python2.5 \
-c hello.c -o build/temp.macosx-10.5-i386-2.5/hello.o

Очевидно, этому не удается найти arrayobject.h. Как я могу заставить distutils использовать корректное местоположение numpy, включают файлы (не заставляя пользователя определить $CFLAGS)?

42
задан Vebjorn Ljosa 4 March 2010 в 14:17
поделиться

1 ответ

Использовать numpy.get_include():

from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
import numpy as np                           # <---- New line

ext_modules = [Extension("hello", ["hello.pyx"])]

setup(
  name = 'Hello world app',
  cmdclass = {'build_ext': build_ext},
  include_dirs = [np.get_include()],         # <---- New line
  ext_modules = ext_modules
)
66
ответ дан 26 November 2019 в 23:38
поделиться
Другие вопросы по тегам:

Похожие вопросы: