无需调用python setup.py或setuptools即可编译Cython扩展

时间:2018-11-02 13:17:49

标签: python setuptools cythonize

我正在使用Docker容器来封装深度学习实验。设置Docker时,我正在运行一个python文件,该文件首先从cythonize运行Cython.Build,然后在编译后的文件上运行gcc

是否可以通过这种方式使用setuptools?例如,https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/setup.py编译CUDA和Cython扩展,然后将它们作为软件包安装。但是,对于我的用例,这很不方便。

基本上,我希望通过此脚本通过setup()之类的函数运行编译,而无需实际安装软件包。

setup(
    name="maskrcnn_benchmark",
    version="0.1",
    author="fmassa",
    url="https://github.com/facebookresearch/maskrnn-benchmark",
    description="object detection in pytorch",
    packages=find_packages(exclude=("configs", "tests",)),
    # install_requires=requirements,
    ext_modules=get_extensions(),
    cmdclass={"build_ext": torch.utils.cpp_extension.BuildExtension},
)

0 个答案:

没有答案