我正在使用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},
)