Difference between revisions of "Tensorflow JNI development"
From ElphelWiki
(Created page with "==About== Based on [https://medium.com/@maxime.durand.54/build-tennsorflow-2-0-for-java-on-windows-2ab51b9cac45 this]. How to: * Install things * Build libtensorflow.jar, li...") |
(No difference)
|
Revision as of 16:08, 4 March 2020
Contents
About
Based on this.
How to:
- Install things
- Build libtensorflow.jar, libtensorflow_jni.so
- Install pom.xml and libtensorflow.jar to local maven repo in ~/.m2
- Modify Tensorflow JNI functions
- Create a maven project and use modified JNI
Install
- In Kubuntu
- Get TF 1.15.0 - git or archive
- Install bazel 0.25.2 - see *.deb in releases
Build
cd ~/git/tensorflow-1.15.0 ./configure bazel build -c opt //tensorflow/java:tensorflow //tensorflow/java:libtensorflow_jni bazel build -c opt //tensorflow/java:pom mvn install:install-file -Dfile=bazel-bin/tensorflow/java/libtensorflow.jar -DpomFile=bazel-bin/tensorflow/java/pom.xml
Artifacts of interest in bazel-bin/tensorflow/java/:
libtensorflow_jni.so libtensorflow.jar pom.xml
- xml and jar are taken care of.
- so will have to be in the library path - set LD_LIBRARY_PATH or PATH or go "java -Djava.library.path"
Modify TF JNI
For example, one wants to create a new function in org.tensorflow.TensorFlow package. Then, see:
tensorflow/java/src/main/java/org/tensorflow/ tensorflow/java/src/main/native/
- add native method to tensorflow/java/src/main/java/org/tensorflow/TensorFlow.java
- add to header file tensorflow/java/src/main/native/tensorflow_jni.h
- add to c file tensorflow/java/src/main/native/tensorflow_jni.cc