open-mmlab / mmtracking

OpenMMLab Video Perception Toolbox. It supports Video Object Detection (VID), Multiple Object Tracking (MOT), Single Object Tracking (SOT), Video Instance Segmentation (VIS) with a unified framework.
https://mmtracking.readthedocs.io/en/latest/
Apache License 2.0
3.59k stars 598 forks source link

mmtracking with torch >= 2.2.2 fails with error: RuntimeError: nms_impl: implementation for device cuda:0 not found. #940

Open nvijayrania opened 3 months ago

nvijayrania commented 3 months ago

Checklist

  1. I have searched related issues but cannot get the expected help.
  2. The bug has not been fixed in the latest version.

Describe the bug There seems to be a circular dependency when I'm using mmtracking with torch 2.2.2 and higher versions.

Bug fix Could we please remove this assertion and make mmtracking compatible with latest mmcv 2.x versions to allow torch version >= 2.x.