Fix GPU builds#385
Conversation
Signed-off-by: Ryan Nett <JNett96@gmail.com>
Signed-off-by: Ryan Nett <JNett96@gmail.com>
|
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 <JNett96@gmail.com>
Signed-off-by: Ryan Nett <JNett96@gmail.com>
|
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
.bazelrcand CI CUDA and CUDNN versions.@karllessard can you tag this for CI build?
I just overwrote the
.bazelrcfile, if there's anything we want to add we should be doing it in a separate file, anyways.