ImportError: /root/userfolder/pic-challenge-baseline/lib/fpn/nms/_ext/nms/_nms.so: undefined symbol: __cudaPopCallConfiguration

[XinJiang1994]

Traceback (most recent call last):
File "models/train.py", line 25, in
from lib.motif_pic import RelModel
File "/root/userfolder/pic-challenge-baseline/lib/motif_pic.py", line 14, in
from lib.fpn.nms.functions.nms import apply_nms
File "/root/userfolder/pic-challenge-baseline/lib/fpn/nms/functions/nms.py", line 4, in
from .._ext import nms
File "/root/userfolder/pic-challenge-baseline/lib/fpn/nms/_ext/nms/init.py", line 3, in
from ._nms import lib as _lib, ffi as _ffi

ImportError: /root/userfolder/pic-challenge-baseline/lib/fpn/nms/_ext/nms/_nms.so: undefined symbol: __cudaPopCallConfiguration

---

I installed torch==0.4.1 and torchvision==0.2.2
Then I run make successfully
I want to use PIC to train. and I run ./scripts/train_motif.sh as the readme said.
Then I got the error above. Do I miss any steps?