-
Notifications
You must be signed in to change notification settings - Fork 216
Fix GPU builds #385
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
Fix GPU builds #385
Conversation
Signed-off-by: Ryan Nett <[email protected]>
Signed-off-by: Ryan Nett <[email protected]>
I'm testing this locally, but it's still building. |
I could copy the tensorflow bazelrc from the remote repo, and allow us to use our own for whatever we need (and prevent this kind of out-of-sync issue from happening again). It would make it much harder to edit though (you'd need to use a patch). Thoughts? |
Signed-off-by: Ryan Nett <[email protected]>
Signed-off-by: Ryan Nett <[email protected]>
This is working on my machine, so we should be good to merge. |
Would be good to figure out where the |
The CI timing out since it can't reuse the cache. |
Yes, so let's merge it, anyway we know that the current build is not working properly on GPU devices, thanks @rnett |
It's timing out on |
Finally got the snapshots to build, I think it took 4 passes this time... you can try it out now |
Fixes GPU builds by updating
.bazelrc
and CI CUDA and CUDNN versions.@karllessard can you tag this for CI build?
I just overwrote the
.bazelrc
file, if there's anything we want to add we should be doing it in a separate file, anyways.