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.

Sign up for GitHub

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

Jump to bottom

Tensorflow GPU #145

Open
ckchow opened this issue Sep 16, 2016 · 0 comments
Open

Tensorflow GPU #145

ckchow opened this issue Sep 16, 2016 · 0 comments

Comments

@ckchow
Copy link

ckchow commented Sep 16, 2016

I've been running a modified version of the ImagenetTF app on the Places dataset, and whenever I check the worker container logs it looks like all of the TF graph nodes are being put on CPU. I've looked at the model defs (.pb and py files) and they are correctly requesting all nodes to be placed on GPU, but the soft-placement constraint is allowing them to failover to CPU I guess.

Does anyone know what could be causing this? Could it be the JavaCPP bindings? I'm using the 03/05 prebuilt GPU jars.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@ckchow