Python setup.py 에서 닌자 관련 에러 날 때

가끔 setuptools에서 이상한 에러가 날 때가 있는데, 주로 컴파일러 문제일 때가 많다.

에러 예시 : nvcc fatal Unknown option '-generate-dependencies-with-compile'

cmake도 에러가 자주 나지만 닌자에서도 에러를 자주 보게 되어서, 닌자랑 cmake랑 둘 다 시도해보고 싶은 상황이 있게 된다.

from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension

setup(
    name='roi_align_cuda',
    ext_modules=[
        CUDAExtension('roi_align_cuda', [
            'src/roi_align_cuda.cpp',
            'src/roi_align_kernel.cu',
        ]),
    ],
    cmdclass={'build_ext': BuildExtension.with_options(use_ninja=False)})

.with_options(use_ninja=False) 로 주면 닌자가 아니라 cmake를 쓴다. 대신에 conda든 apt-get이든 cmake는 깔려있어야 함!