admin管理员组

文章数量:1405585

I am currently trying to reproduce DN-DETR on an NVIDIA RTX 5090 GPU. My environment details are:

  • GPU: NVIDIA RTX 5090
  • PyTorch version: 2.8.0.dev20250322+cu128 (nightly)
  • CUDA version: 12.8
  • OS: Ubuntu (assuming Linux-based system)

When I try to import MultiScaleDeformableAttention from models/dn_dab_deformable_detr/ops, I get the following error:

(/home/user1/thanhpd) user1@xbk1:~/thanhpd/code/implementation-of-dn-detr/DN-DETR/models/dn_dab_deformable_detr/ops$ python -c "import MultiScaleDeformableAttention"
Traceback (most recent call last):
  File "<string>", line 1, in <module>
ImportError: /home/user1/thanhpd/code/implementation-of-dn-detr/DN-DETR/models/dn_dab_deformable_detr/ops/MultiScaleDeformableAttention.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c106detail23torchInternalAssertFailEPKcS2_jS2_RKSs

Any suggestions i can try?

What I Have Tried:

  • Recompiling the MultiScaleDeformableAttention :
!python setup.py build install
  • Try 
export CUDA_HOME=/usr/local/cuda-12.8
  • Tried numerous different versions of pytorch and torchvision, but because my GPU is 50x0 cards, so only nighty version of pytorch supports it.

本文标签：

版权声明：本文标题：python - MultiScaleDeformableAttention.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c106detail23torchInternalAssertFai 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.betaflare.com/web/1744259304a2597633.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。