Giters

roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

undefined symbol: __cudaRegisterFatBinaryEnd

ZahraAnam opened this issue · comments

commented

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

  • Operating system: Linux20.04
  • CUDA version: 10.0
  • cuDNN version: 7.5
  • GPU models (for all devices if they are not all the same): RTX
  • python version: 3.6.6
  • pytorch version: 0.3.1
  • Anything else that seems relevant: ?
commented

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