roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 567 forks source link

undefined symbol: __cudaRegisterFatBinaryEnd #237

Open ZahraAnam opened 3 years ago

ZahraAnam commented 3 years ago

I am getting the following error and I don't understand how to get rid of it coz my cuda version and symlinks are correct.

ImportError: /home/anam/Codes/mask-rcnn.pytorch/lib/model/roi_pooling/_ext/roi_pooling/_roi_pooling.so: undefined symbol: __cudaRegisterFatBinaryEnd

Command Ran is just a try to run inference

python tools/infer_simple.py

System information

priscillalui commented 3 years ago

I have the same issue. Any insight would be appreciated.