undefined symbol: _Z27ms_deform_attn_cuda_forwardRKN2at6TensorES2_S2_S2_S2_i

[Adamscc]

suceed in "sh make.sh", but when importing MultiScaleDeformableAttention, it crashed

Traceback (most recent call last):
File "main.py", line 34, in
main()
File "main.py", line 23, in main
run_alfred()
File "/home/adamsc/HELPER/models/alfred_eval_embodied_llm.py", line 1224, in run_alfred
subgoalcontroller = SubGoalController(
File "/home/adamsc/HELPER/models/alfred_eval_embodied_llm.py", line 297, in init
self.object_tracker = ObjectTrack(
File "/home/adamsc/HELPER/task_base/object_tracker.py", line 79, in init
from nets.solq import DDETR
File "/home/adamsc/HELPER/nets/solq.py", line 16, in
from SOLQ.models import build_model
File "/home/adamsc/HELPER/SOLQ/models/init.py", line 12, in
from .solq_base import build as build_solq
File "/home/adamsc/HELPER/SOLQ/models/solq_base.py", line 30, in
from .deformable_transformer_solq_base import build_deforamble_transformer
File "/home/adamsc/HELPER/SOLQ/models/deformable_transformer_solq_base.py", line 21, in
from SOLQ.models.ops.modules import MSDeformAttn
File "/home/adamsc/HELPER/SOLQ/models/ops/modules/init.py", line 11, in
from .ms_deform_attn import MSDeformAttn
File "/home/adamsc/HELPER/SOLQ/models/ops/modules/ms_deform_attn.py", line 23, in
from ..functions import MSDeformAttnFunction
File "/home/adamsc/HELPER/SOLQ/models/ops/functions/init.py", line 11, in
from .ms_deform_attn_func import MSDeformAttnFunction
File "/home/adamsc/HELPER/SOLQ/models/ops/functions/ms_deform_attn_func.py", line 20, in
import MultiScaleDeformableAttention as MSDA
ImportError: /home/adamsc/.local/lib/python3.8/site-packages/MultiScaleDeformableAttention-1.0-py3.8-linux-x86_64.egg/MultiScaleDeformableAttention.cpython-38-x86_64-linux-gnu.so: undefined symbol: _Z27ms_deform_attn_cuda_forwardRKN2at6TensorES2_S2_S2_S2_i

my cuda version: 12.1
my torch: 2.4.1+cu121
my device : ubuntu 24.04 LTS

[n-shamsi]

In my experience, it seems that using a different pytorch-cuda version helps resolves undefined symbol error.