No module named 'tree_filter_cuda'

[matanPercepto]

Hi,
I tried installing your package inside an nvidia cuda docker, I made sure to install correct and cuda supporting versions of pytorch and mmcv-full, and used the bash setup.sh command to install your package, but when I tried to run it, I receive the following error:
File "/*****.py", line 5, in <module> from mmdet.apis import init_detector File "/app/BoxInstSeg/mmdet/apis/__init__.py", line 2, in <module> from .inference import (async_inference_detector, inference_detector, File "/app/BoxInstSeg/mmdet/apis/inference.py", line 13, in <module> from mmdet.datasets import replace_ImageToTensor File "/app/BoxInstSeg/mmdet/datasets/__init__.py", line 9, in <module> from .utils import (NumClassCheckHook, get_loading_pipeline, File "/app/BoxInstSeg/mmdet/datasets/utils.py", line 11, in <module> from mmdet.models.roi_heads.mask_heads import FusedSemanticHead File "/app/BoxInstSeg/mmdet/models/__init__.py", line 7, in <module> from .dense_heads import * # noqa: F401,F403 File "/app/BoxInstSeg/mmdet/models/dense_heads/__init__.py", line 5, in <module> from .box_solov2_head import BoxSOLOv2Head File "/app/BoxInstSeg/mmdet/models/dense_heads/box_solov2_head.py", line 13, in <module> from mmdet.ops.tree_filter.modules.tree_filter import MinimumSpanningTree, TreeFilter2D File "/app/BoxInstSeg/mmdet/ops/tree_filter/modules/tree_filter.py", line 5, in <module> from ..functions.mst import mst File "/app/BoxInstSeg/mmdet/ops/tree_filter/functions/mst.py", line 7, in <module> import tree_filter_cuda as _C ModuleNotFoundError: No module named 'tree_filter_cuda'
Any advice on what I did wrong in the installation?
Thanks!

system details:
CUDA: 11.1.1
cudnn: 8
ubuntu: 20.04
python 3.9
pytorch 1.10.1
mmcv-full: 1.5.0

[LiWentomng]

@matanPercepto Hello, please try to use pytorch1.9.0 under CUDA 11.1.1, and re-compile tree_filter_cuda operation by the following.

cd mmdet/ops/tree_filter 
python setup.py build develop

[matanPercepto]

Thanks, I tried it but now the installation fails with error:
Installed /app/BoxInstSeg/mmdet/ops/pairwise #20 66.77 Processing dependencies for boxinst-plugin==0.0.1 #20 66.77 Finished processing dependencies for boxinst-plugin==0.0.1 #20 68.03 No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda' #20 68.03 Traceback (most recent call last): #20 68.03 File "/app/BoxInstSeg/mmdet/ops/tree_filter/setup.py", line 22, in <module> #20 68.03 raise Exception('This implementation is only avaliable for CUDA devices.') #20 68.03 Exception: This implementation is only avaliable for CUDA devices.
I tried adding ENV CUDA_VISIBLE_DEVICES=0 and ENV CUDA_HOME=/usr/local/cuda but it still didn't work.