Trouble Installing mmdet and mmrotate on Jetson AGX Orin

[sznt.norbi]

Hello everyone,

I’m having some trouble setting up mmdet on my Jetson AGX Orin device, which is running JetPack 6.0. Initially, I tried using the dustynv/l4t-pytorch:r36.2.0 Docker image, but I encountered issues when I attempted to install mmdet and mmrotate.

I’m also trying to install mmdet and mmrotate directly on the system with pytorch 2.4.0. Everything goes smoothly until I reach the point where I try to install mmdet, at which point I receive the following error: Building wheel for mmcv (setup.py) ... error

I’m not sure how to resolve this issue and would appreciate any advice or tips from anyone who has successfully installed mmdet on a similar setup. Has anyone else faced this problem on a Jetson device? Are there any specific dependencies or steps I might be missing?

Thank you in advance for your help!

[AastaLLL]

Hi,

Is the error message only Building wheel for mmcv (setup.py) ... error?
If there are more logs, could you share them with us as well?

We have tested the installation on our device with JetPack 6.0 and it works well:

$ pip3 install mmdet
Defaulting to user installation because normal site-packages is not writeable
Collecting mmdet
  Downloading mmdet-3.3.0-py3-none-any.whl (2.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.2/2.2 MB 2.5 MB/s eta 0:00:00
Requirement already satisfied: matplotlib in /usr/lib/python3/dist-packages (from mmdet) (3.5.1)
Collecting tqdm
  Downloading tqdm-4.66.5-py3-none-any.whl (78 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 78.4/78.4 KB 316.3 kB/s eta 0:00:00
Collecting terminaltables
  Downloading terminaltables-3.1.10-py2.py3-none-any.whl (15 kB)
Collecting pycocotools
  Downloading pycocotools-2.0.8-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (410 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 410.6/410.6 KB 757.3 kB/s eta 0:00:00
Requirement already satisfied: numpy in /usr/lib/python3/dist-packages (from mmdet) (1.21.5)
Requirement already satisfied: scipy in /usr/lib/python3/dist-packages (from mmdet) (1.8.0)
Collecting shapely
  Downloading shapely-2.0.5-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (2.4 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.4/2.4 MB 1.1 MB/s eta 0:00:00
Requirement already satisfied: six in /usr/lib/python3/dist-packages (from mmdet) (1.16.0)
Installing collected packages: tqdm, terminaltables, shapely, pycocotools, mmdet
Successfully installed mmdet-3.3.0 pycocotools-2.0.8 shapely-2.0.5 terminaltables-3.1.10 tqdm-4.66.5

Thanks.