Error importing MultiScaleDeformableAttention

[jtaylorbus]

I ran the make.sh script in models/ops/ and it completed without error, but when I try to run inference I get the following error when importing MultiScaleDeformableAttention:

ImportError: /opt/conda/lib/python3.10/site-packages/MultiScaleDeformableAttention-1.0-py3.10-linux-x86_64.egg/MultiScaleDeformableAttention.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN2at4_ops5zeros4callEN3c108ArrayRefINS2_6SymIntEEENS2_8optionalINS2_10ScalarTypeEEENS6_INS2_6LayoutEEENS6_INS2_6DeviceEEENS6_IbEE

I've tried multiple pytorch Docker images and re-ran make.sh for each, but still get the same error every time. Can you please clarify what base image is required or provide one with the repo?

[HELLORPG]

Have you tried to run the test.py in models/ops/? And what's your CUDA version?

[HELLORPG]

Have you resolved this problem? If there are no updates, I will close this issue in the near future.

[HELLORPG]

I temporarily closed this issue. If you have any further questions about this problem, you can re-open this issue.