Running continuously outputs' it is running on CPU! 'NoneType object has no attribute' ms_deform_attn_forward ''

naver-ai / egtr

[CVPR 2024 Best paper award candidate] EGTR: Extracting Graph from Transformer for Scene Graph Generation

https://arxiv.org/abs/2404.02072

Apache License 2.0

78 stars 2 forks source link

Open wuzhiwei2001 opened 1 month ago

wuzhiwei2001 commented 1 month ago

I am running code on an A100 and outputting this situation：微信图片_20241023171104

wuzhiwei2001 commented 1 month ago

Looking forward to your reply, I would greatly appreciate it

ha1ha2hahaha commented 4 weeks ago

I think you should check ninja in your device