MultiScaleDeformableAttention import error

[remeliashirlley]

I'm using Python 3.8 with PyTorch 1.9 and Cuda 11.1. I have already set cuda_home as such:

export CUDA_HOME=/mnt/slurm_home/remelias/anaconda3/envs/vita30/
cd /mnt/slurm_home/remelias/MeViS-main/mask2former/modeling/pixel_decoder/ops/
sh make.sh

I'm still getting MSDA import error.

Traceback (most recent call last):
line 22, in
import MultiScaleDeformableAttention as MSDA
ImportError: /mnt/slurm_home/remelias/anaconda3/envs/vita30/lib/python3.8/site-packages/MultiScaleDeformableAttention-1.0-py3.8-linux-x86_64.egg/MultiScaleDeformableAttention.cpython-38-x86_64-linux-gnu.so: undefined symbol: _ZNK2at10TensorBase8data_ptrIdEEPT_v

How can I resolve this?

[heshuting555]

It maybe a Pytorch-CUDA version mismatch problem (see here). You may upgrade your PyTorch to 2.0. Our code is based on PyTorch2.0.1+cu117.

[remeliashirlley]

Thank you. Using PyTorch 2.0.1+cu117 resolved the error.