No module named 'mmdet.ops.nms.gpu_nms'

[JR-Wang]

When I run 'python3 tools/train.py configs/masktrack_rcnn_r50_fpn_1x_youtubevos.py', an error occured as follow: File "tools/train.py", line 7, in from mmdet.datasets import get_dataset File "/home/lin/python_work/competition/OVIS/MaskTrackRCNN-master/mmdet/datasets/init.py", line 1, in from .custom import CustomDataset File "/home/lin/python_work/competition/OVIS/MaskTrackRCNN-master/mmdet/datasets/custom.py", line 11, in from .extra_aug import ExtraAugmentation File "/home/lin/python_work/competition/OVIS/MaskTrackRCNN-master/mmdet/datasets/extra_aug.py", line 5, in from mmdet.core.evaluation.bbox_overlaps import bbox_overlaps File "/home/lin/python_work/competition/OVIS/MaskTrackRCNN-master/mmdet/core/init.py", line 6, in from .post_processing import * # noqa: F401, F403 File "/home/lin/python_work/competition/OVIS/MaskTrackRCNN-master/mmdet/core/post_processing/init.py", line 1, in from .bbox_nms import multiclass_nms File "/home/lin/python_work/competition/OVIS/MaskTrackRCNN-master/mmdet/core/post_processing/bbox_nms.py", line 3, in from mmdet.ops.nms import nms_wrapper File "/home/lin/python_work/competition/OVIS/MaskTrackRCNN-master/mmdet/ops/init.py", line 1, in from .nms import nms, soft_nms File "/home/lin/python_work/competition/OVIS/MaskTrackRCNN-master/mmdet/ops/nms/init.py", line 1, in from .nms_wrapper import nms, soft_nms File "/home/lin/python_work/competition/OVIS/MaskTrackRCNN-master/mmdet/ops/nms/nms_wrapper.py", line 4, in from .gpu_nms import gpu_nms ModuleNotFoundError: No module named 'mmdet.ops.nms.gpu_nms'

How to fix this error? Many thanks.

[jiojio0505]

I have the same problem and haven't found a solution.

[zhw2020913]

You maybe not execute PYTHON=python3 ./compile.sh. here https://github.com/open-mmlab/mmdetection/issues