A problem about nms

Hi Rowan, thanks for your codes sharing. However, I find nms.nms_apply(keep, boxes_sorted, nms_thresh) in lib/fpn/nms/functions/nms.py would make pytorch break up. Before I execute that line, the operation of Tensor.cuda()+int() is legal, but after I execute that line this operation will raise error: RuntimeError: cuda runtime error (48) : no kernel image is available for execution on the device at /pytorch/torch/lib/THC/generic/THCTensorMathPairwise.cu:21 when it goes on keep_im + s at line 24. Could you please help me? Thanks!

[alibabadoufu]

For those who are still solving this issues, here are the steps you should consider to follow:

1. Change SM_# (at places listed below) that suits your current GPU cards (see this for your reference)

• lib/lstm/highway_lstm_cuda/make.sh (line 15)
• lib/fpn/roi_align/src/cuda/Makefile (line 2)
• lib/fpn/nms/src/cuda/Makefile (line 2)

1. Afterwards, you should also delete *.o files (at places listed below) before you try recompiling nms, roi_align and highway_lstm)

• lib/lstm/highway_lstm_cuda/src/highway_lstm_kernel.cu.o
• lib/fpn/roi_align/src/cuda/roi_align.cu.o
• lib/fpn/nms/src/cuda/nms.cu.o

1. Run ‘make’ in your home directory.

That’s all you need to do to change your faulty SM setup to the correct one.

[YiwuZhong]

Hi, I’ve met the same problem using Tesla P100. So did you solve your problem only changing from _61 to _60 in 3 files (nms, roi_align and highway_lstm_cuda)? And you said “replaced nms and roi_align with other’s codes”， could you please tell what “other’s codes” did you use to solve the problem? Thanks. 😄