Skip to content

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