demo_coco.py:undefined symbol: __cudaPopCallConfiguration
MoFeiHu opened this issue · comments
Hi! I have built nms and ROIAlign successfully, but when I run demo_coco.py, the Error occurs:
File "/home/hmf/Downloads/Deep_Learning/Mask_RCNN_Pytorch-master/demo_coco.py", line 10, in
import coco
File "/home/hmf/Downloads/Deep_Learning/Mask_RCNN_Pytorch-master/coco.py", line 50, in
import model as modellib
File "/home/hmf/Downloads/Deep_Learning/Mask_RCNN_Pytorch-master/model.py", line 26, in
from nms.nms_wrapper import nms
File "/home/hmf/Downloads/Deep_Learning/Mask_RCNN_Pytorch-master/nms/nms_wrapper.py", line 11, in
from nms.pth_nms import pth_nms
File "/home/hmf/Downloads/Deep_Learning/Mask_RCNN_Pytorch-master/nms/pth_nms.py", line 2, in
from ._ext import nms
File "/home/hmf/Downloads/Deep_Learning/Mask_RCNN_Pytorch-master/nms/_ext/nms/init.py", line 3, in
from ._nms import lib as _lib, ffi as _ffi
ImportError: /home/hmf/Downloads/Deep_Learning/Mask_RCNN_Pytorch-master/nms/_ext/nms/_nms.so: undefined symbol: __cudaPopCallConfiguration
How can I solve this problem? Any suggestion will be thankful!
it's from the misalignment between pytorch version and cuda version
see here for detail jwyang/faster-rcnn.pytorch#190