Unable to import _C from detectron2 inside Docker container

[alwibowo]

Hello,

I have this weird import error when trying to install Detectron2 inside a docker container.

Context

I basically built detectron2 from the source and pushed it into a private repository in packagecloud. I built it by using the following specific python and torch versions:

• python: 3.10.13
• torch: 2.2.1

Problem

This approach actually works well locally. I currently use M1 Pro MacOs and manage to install detectron2 from the packagecloud. My model runs without any problem. But when I try to install it inside docker container, this is where the problem starts. I encounter the following error:

ImportError: /app/.venv/lib/python3.10/site-packages/detectron2/_C.cpython-310-aarch64-linux-gnu.so: undefined symbol: _ZN5torch3jit17parseSchemaOrNameERKSsb

I tried to import _C from detectron2 locally on my M1 Pro and it worked. Has anyone encountered this before and managed to make it work in a docker container?