search

megvii-research / FSCE

Apache License 2.0
275 stars 47 forks source link

runtinmeError:Not compiled with GPU support #16

Closed john2020-210 closed 3 years ago

john2020-210 commented 3 years ago

When I run the code ,I met the following problem" RuntinmeError:Not compiled with GPU support " dose anyone meet the same problem? my environment is pytorch1.71 cuda 10.1 GRX 2080Ti

bsun0802 commented 3 years ago

Try pytorch 1.4, and build again.

To get started, maybe you can start with fsdet(https://github.com/ucbdrive/few-shot-object-detection) to see if you can get it to run first.

john2020-210 commented 3 years ago

@bsun0802 Thanks for your reply.I have tried with pytorch1.4 and started the fsdet successfully ,but it still report the RuntimeError:Not compiled with GPU support(ROIAlign_forward at FSCE-main /fsdet/layers/csrc/ROIAlign/ROIAlign.h:73

bsun0802 commented 3 years ago

Maybe you can delete it and try build again? I'm no expert in build, ours use the same build code from fsdet.

It seems other people were able to run our code. Maybe you can find information in other issues?

john2020-210 commented 3 years ago

Thank you all the same，I find the error mainly relates to the ROIAlign_forward in fsdet/layers/csrc/ROIAlign/ROIAlign.73,could you help me ?

bsun0802 commented 3 years ago

ROIAlign as you mentioned, are C source files and I believe coming from detectron2, and we do not modify and we believe fsdet's author will not modify ROI Align as well.

Sorry but I have no idea bro haha.