search

henghuiding / MeViS

[ICCV 2023] MeViS: A Large-scale Benchmark for Video Segmentation with Motion Expressions
https://henghuiding.github.io/MeViS/
MIT License
485 stars 19 forks source link

MultiScaleDeformableAttention import error #9

Open remeliashirlley opened 8 months ago

remeliashirlley commented 8 months ago

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 commented 8 months ago

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 commented 8 months ago

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