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-only support #6

Open
zplizzi opened this issue Nov 3, 2016 · 2 comments
Open

CPU-only support #6

zplizzi opened this issue Nov 3, 2016 · 2 comments

Comments

@zplizzi
Copy link
Owner

zplizzi commented Nov 3, 2016

Copied from #2.

I got a problem when compiling the roi_pooling_op. Since I do not have GPU installed, I have configured bazel using cpu only. But when I tried to load the lib from python repl, it says:

tensorflow.python.framework.errors.NotFoundError: dlopen(/Users/dtong/code/data/tensorflow/bazel-bin/tensorflow/core/user_ops/roi_pooling_op_grad.so, 6): Symbol not found: __Z27ZeroTensorGpuKernelLauncherPfi

Do you have any idea of bazel configuration that could ignore the gpu code?

@zplizzi
Copy link
Owner Author

zplizzi commented Nov 3, 2016

Unfortunately, I don't think CPU-only compilation will work with the code as-is. I tried to share some code between the two versions, and it looks like that's breaking CPU-only compatibility. If you need this, you can probably just slightly modify the code to remove the GPU parts entirely. It'd probably be slightly more difficult to better separate the versions such that it can be compiled either way. I'd definitely welcome any contributions to that regard, though!

@tongda
Copy link

tongda commented Nov 4, 2016

Yes, I understand. I have just commented lines related to GPU code. Thanks.

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