search

longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.55k stars 420 forks source link

there is no cython_bbox #81

Open masoudpz opened 6 years ago

masoudpz commented 6 years ago

hi, in derknet.py line 9 No module named 'utils.cython_bbox' there is no utils.cython_bbox in files, how can i find this file?

longcw commented 6 years ago

Did you run make.sh successfully?

masoudpz commented 6 years ago

Traceback (most recent call last): File "build.py", line 66, in CUDA = locate_cuda() File "build.py", line 61, in locate_cuda 'be located in %s' % (k, v)) OSError: The CUDA lib64 path could not be located in /usr/lib64 Compiling reorg layer kernels by nvcc... Including CUDA code. generating /tmp/tmpr0_ov0fi/_reorg_layer.c running build_ext building '_reorg_layer' extension

make.sh output

longcw commented 6 years ago

It seems utils/build.py cannot find you cuda lib https://github.com/longcw/yolo2-pytorch/blob/master/utils/build.py#L29 You can set it in the code or in env CUDAHOME.