search

longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.54k stars 421 forks source link

undefined symbol: __cudaUnregisterFatBinary #63

Open suyashnigam opened 6 years ago

suyashnigam commented 6 years ago

Having this error after successfully executing the make shell file. File "demo.py", line 6, in from darknet import Darknet19 File "/home/suyash/AIML/yolo2-pytorch/darknet.py", line 8, in from layers.reorg.reorg_layer import ReorgLayer File "/home/suyash/AIML/yolo2-pytorch/layers/reorg/reorg_layer.py", line 3, in from ._ext import reorg_layer File "/home/suyash/AIML/yolo2-pytorch/layers/reorg/_ext/reorg_layer/init.py", line 3, in from ._reorg_layer import lib as _lib, ffi as _ffi ImportError: /home/suyash/AIML/yolo2-pytorch/layers/reorg/_ext/reorg_layer/_reorg_layer.so: undefined symbol: __cudaUnregisterFatBinary

Ubuntu: 16.04 Cuda: 9.1 nvcc: 7.5 Pytorch: 0.3.1