search

ShuangXieIrene / ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.
MIT License
568 stars 165 forks source link

No module named 'lib.utils.nms._ext.nms._nms' #2

Closed fengyang95 closed 6 years ago

fengyang95 commented 6 years ago

When I try to run the demo, I get the following error: from ._nms import lib as _lib, ffi as _ffi ImportError: No module named 'lib.utils.nms._ext.nms._nms'

Ubuntu 16.04.3 GTX 1080Ti Cuda 8.0 installed

foreverYoungGitHub commented 6 years ago

Currently there are two nms exists in the code. One is implemented by pytorch, the other is implemented. For the easier usage. I'll change all the nms to pytorch version. But if you want to use the cuda version, you need to compile the code based on this . The original version is compiled by Ubuntu 16.04.3, Cuda 9.0.

Besides, my personal suggest is that, the new cuda version have good improvement in speed.