_C.roi_align_forward raises a "RuntimeError: Not compiled with GPU support"

ucbdrive / few-shot-object-detection

Implementations of few-shot object detection benchmarks

Apache License 2.0

1.08k stars 225 forks source link

_C.roi_align_forward raises a "RuntimeError: Not compiled with GPU support" #122

Closed E-Galois closed 2 years ago

E-Galois commented 3 years ago

I'm running the demo.py script, but from file "fsdet\layers\roi_align.py", the "_C.roi_align_forward" method raises a "RuntimeError: Not compiled with GPU support". My pytorch is consistent with the cuda version 11.1 and I've already built fsdet with "python setup.py build develop".

I tried to run demo in detectron2, but it does not raise such an error, and can finish inference. What can be the cause?

thomasehuang commented 2 years ago

See this. The detectron2 version we use is an old version (v0.2.1) and may not be compatible with higher CUDA versions. We plan on upgrading to the newest detectron2 version, but this may take some time.

thomasehuang commented 2 years ago

Just updated the installation instructions, you can try following it again. Will close this for now, but feel free to reopen it if it's still an issue for you.