A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
2.82k
stars
567
forks
source link
undefined symbol: __cudaRegisterFatBinaryEnd #237
Open
ZahraAnam opened 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
System information