Error when running test_nms.py

potterhsu / easy-faster-rcnn.pytorch

An easy implementation of Faster R-CNN (https://arxiv.org/pdf/1506.01497.pdf) in PyTorch.

MIT License

165 stars 57 forks source link

Error when running test_nms.py #10

Closed peiherng97 closed 5 years ago

peiherng97 commented 5 years ago

easy-faster-rcnn.pytorch/support/_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _Z21ROIAlign_forward_cudaRKN2at6TensorES2_fiii

I have this error when running test_nms.py Any ideas why this is happening? I've tried searching for relevant issues online but they do not have this kind of error

peiherng97 commented 5 years ago

Managed to solve this issue