Skip to content

Build the tensorflow_cc target instead of just tensorflow #113

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

Merged
merged 1 commit into from
Sep 13, 2020

Conversation

saudet
Copy link
Contributor

@saudet saudet commented Sep 12, 2020

Fixes #82

All tests pass on full builds:
https://github.com/saudet/tensorflow-java/actions/runs/250665886
Does not increase total build time when compared with current ones like:
https://github.com/tensorflow/java/actions/runs/215892440

Also:

  • Install CUDA via RPM files since NVIDIA broke their Docker containers
  • Build for 3.5 and 7.0 and keep PTX code (fixes TF Java 2.3 failed on Tesla V100 GPU #104)
  • Add "tensorflow/core/util/port.h" to the functions exposed via JavaCPP

/cc @roywei

@karllessard karllessard merged commit 8c67525 into tensorflow:master Sep 13, 2020
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.

TF Java 2.3 failed on Tesla V100 GPU Cannot run model using tensorflow-text ops
2 participants