search

longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.54k stars 421 forks source link

cpu only #13

Closed kaishijeng closed 7 years ago

kaishijeng commented 7 years ago

Plan to run this on Raspberry3 board. Does you code run on a CPU only platform?

Thanks,

9thDimension commented 7 years ago

I just ran the demo successfully on a laptop with no nvidia. Comment out lines like https://github.com/longcw/yolo2-pytorch/blob/master/demo.py#L34 and https://github.com/longcw/yolo2-pytorch/blob/master/demo.py#L47 which place tensors onto the GPU memory and you should be OK.

kaishijeng commented 7 years ago

Thanks for the info. But the compilation make.sh can't be completed because it needs nvcc to compile roi_pooling. Also utils needs nvcc in order to build. How do you solve this?

AsWali commented 6 years ago

@kaishijeng Can you tell me how you solved it ?

imransalam commented 6 years ago

I have trained a network on GPU but I have to deploy my code on a CPU instance. The test.py script does not seem to work on CPU and gives 'ImportError: No module named _reorg_layer' error on from ._reorg_layer import lib as _lib, ffi as _ffi. This is probably because make.sh could not be compiled. Any idea how to achieve this? @kaishijeng @9thDimension @longcw

georgeokelly commented 6 years ago

hi, how do you solve it?

qq1243196045 commented 3 years ago

I also want use it with pytorch-cpuonly,What should I do?