search

tianzhi0549 / FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)
https://arxiv.org/abs/1904.01355
Other
3.28k stars 630 forks source link

undefined symbol: cudaSetupArgument #365

Open zhealther opened 3 years ago

zhealther commented 3 years ago

(FCOS) whealther@whealther:~/FCOS$ python demo/fcos_demo.py Traceback (most recent call last): File "demo/fcos_demo.py", line 6, in from predictor import COCODemo File "/home/whealther/FCOS/demo/predictor.py", line 6, in from fcos_core.modeling.detector import build_detection_model File "/home/whealther/FCOS/fcos_core/modeling/detector/init.py", line 2, in from .detectors import build_detection_model File "/home/whealther/FCOS/fcos_core/modeling/detector/detectors.py", line 2, in from .generalized_rcnn import GeneralizedRCNN File "/home/whealther/FCOS/fcos_core/modeling/detector/generalized_rcnn.py", line 11, in from ..backbone import build_backbone File "/home/whealther/FCOS/fcos_core/modeling/backbone/init.py", line 2, in from .backbone import build_backbone File "/home/whealther/FCOS/fcos_core/modeling/backbone/backbone.py", line 7, in from fcos_core.modeling.make_layers import conv_with_kaiming_uniform File "/home/whealther/FCOS/fcos_core/modeling/make_layers.py", line 10, in from fcos_core.layers import Conv2d File "/home/whealther/FCOS/fcos_core/layers/init.py", line 10, in from .nms import nms, ml_nms File "/home/whealther/FCOS/fcos_core/layers/nms.py", line 3, in from fcos_core import _C ImportError: /home/whealther/FCOS/fcos_core/_C.cpython-37m-x86_64-linux-gnu.so: undefined symbol: cudaSetupArgument what can i do?