Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

CPU Mode? #81

Open
KojiShiono opened this issue Jun 20, 2017 · 1 comment
Open

CPU Mode? #81

KojiShiono opened this issue Jun 20, 2017 · 1 comment

Comments

@KojiShiono
Copy link

Hello, I would like to use this in CPU-only environment, and I was following this page but still getting following error upon build.
make.sh: line 9: nvcc: command not found
clang: error: no such file or directory: 'roi_pooling_op.cu.o'
make.sh: line 24: nvcc: command not found
clang: error: no such file or directory: 'psroi_pooling_op.cu.o'

Apparently, this is how it is configured in make.sh, and I am wondering if there is any way I can build this for CPU-only environment.

Thank you in advance!

@LouieYang
Copy link

LouieYang commented Nov 28, 2017

For roi_pooling_op, just follow this repo, which could work in this repo too. For psroi_pooling_op, I think it is not used for some network like VGGnet_test, so you could just skip compiling it and annotate some sentences in lib/networks/init.py (just leave your testing network and annotate others) and lib/networks/network.py (annotate psroi_layer).

I think there are also some lines of code that should be annotated, just follow your make error and things could be done.

Btw, it takes 12-13s to run the forwarding for my trained model on 2014 macOS CPU mode. So, I think running it on GPU is a better way.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants