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

Added an extra system include #3

Merged
merged 2 commits into from
Jul 25, 2017
Merged

Added an extra system include #3

merged 2 commits into from
Jul 25, 2017

Conversation

petewarden
Copy link

No description provided.

@petewarden
Copy link
Author

Looks like we need one more include

@petewarden
Copy link
Author

I also just added support for the Pi Two/Three with NEON by default to the build script. This involves an update to Eigen, the fix for tensorflow#9697

I think we should build both Pi One/Zero and Two/Three wheels, but default to the NEON-enabled ones since they're an order of magnitude faster, and most developers will be using a Pi that's capable.

I tested the two wheels by running:

import tensorflow as tf; sess = tf.Session();
tf.matmul(tf.zeros([1000,1000]), tf.zeros([1000, 1000])).eval(session=sess)

and timing the matmul call. On my Pi Three, the non-NEON install took seven seconds to run that call, but only about 500ms with the NEON wheel installed.

Completed wheels are available for testing at https://drive.google.com/file/d/0B7tFa8eBGIw-V0xKMG5Zem5rczQ/view?usp=sharing

There's a gist with my steps for running the build process on a clean Ubuntu 14.04 Docker image here (which I'm hoping we can turn into a CI build script to run nightly):

https://gist.github.com/petewarden/6aa9650fe72d88889d24efd76598f8cb

],
sha256 = "ca7beac153d4059c02c8fc59816c82d54ea47fe58365e8aded4082ded0b820c4",
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

is this change necessary / will it have side effects?

@ebrevdo ebrevdo merged commit a74bded into ebrevdo:tf_master Jul 25, 2017
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

Successfully merging this pull request may close these issues.

2 participants