OpenMMLab-IoUNet
OpenMMLab-IoUNet copied to clipboard
RuntimeError: Error building extension '_prroi_pooling'
Thanks for your great work. I want to know what version your CUDA and Torch versions are. When I used torch1.9 with cuda 11.1, error shows "subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1".


Hi @YangYangGirl Here is the env for this IoUNet project: NVCC: Cuda compilation tools, release 10.1, V10.1.105 GCC: gcc (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0 PyTorch: 1.5.0+cu101 CuDNN 7.6.3 TorchVision: 0.6.0+cu101 OpenCV: 4.5.2 MMCV: 1.3.8 MMCV Compiler: GCC 7.3 MMCV CUDA Compiler: 10.1 MMDetection: 2.14.0+
I think it is due to the fact that your ninja is not installed. Can you try "pip install ninja"?