-
Notifications
You must be signed in to change notification settings - Fork 19
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
MultiScaleDeformableAttention import error #9
Comments
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. |
Thank you. Using PyTorch 2.0.1+cu117 resolved the error. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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?
The text was updated successfully, but these errors were encountered: