ModuleNotFoundError: No module named 'nms._ext.nms._nms'

在运行PlaneRcnn-master时,运行报错,环境是WIN10+pytorch0.4.1+python3.6+cu90

D:\anaconda\envs\py36\python.exe D:/PlaneRCNN-master/PlaneRCNN-master/evaluate_demo.py
Traceback (most recent call last):
  File "D:/PlaneRCNN-master/PlaneRCNN-master/evaluate_demo.py", line 21, in <module>
    from models.model import *
  File "D:\PlaneRCNN-master\PlaneRCNN-master\models\model.py", line 23, in <module>
    from nms.nms_wrapper import nms
  File "D:\PlaneRCNN-master\PlaneRCNN-master\nms\nms_wrapper.py", line 11, in <module>
    from nms.pth_nms import pth_nms
  File "D:\PlaneRCNN-master\PlaneRCNN-master\nms\pth_nms.py", line 2, in <module>
    from ._ext import nms
  File "D:\PlaneRCNN-master\PlaneRCNN-master\nms\_ext\nms\__init__.py", line 3, in <module>
    from ._nms import lib as _lib, ffi as _ffi
ModuleNotFoundError: No module named 'nms._ext.nms._nms'

进程已结束,退出代码为 1


是编译出了问题还是版本等其他问题啊

可能是版本原因造成的,需要cuda9.0或cuda9.1,同时pytorch的版本应该和cuda相匹配,参考: No module named 'model.nms._ext.nms._nms' · Issue #1 · frank-xwang/towards-universal-object-detection · GitHub Hello, I have tried your script. Thanks for sharing. It reports the error when I run scripts "sh scripts/train_universal.sh " Error log is File "universal_model.py", line 32, in from model.faster_rcnn.resnet_uni import resnet File "/root... https://github.com/frank-xwang/towards-universal-object-detection/issues/1