TensorFlow menyediakan Java API - berguna untuk memuat model yang dibuat dengan Python dan menjalankannya dalam aplikasi Java.
Paket Java Nightly Libtensorflow
Paket Libtensorflow JNI dibuat setiap malam dan diunggah ke GCS untuk semua platform yang didukung. Mereka diunggah ke ember GCS libtensorflow-nightly dan diindeks oleh sistem operasi dan tanggal dibuat.
Platform yang Didukung
TensorFlow untuk Java didukung pada sistem berikut:
- Ubuntu 16.04 atau lebih tinggi; 64-bit, x86
- macOS 10.12.6 (Sierra) atau lebih tinggi
- Windows 7 atau lebih tinggi; 64-bit, x86
Untuk menggunakan TensorFlow di Android, lihat TensorFlow Lite
TensorFlow dengan Apache Maven
Untuk menggunakan TensorFlow dengan Apache Maven , tambahkan dependensi ke file pom.xml
proyek:
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>tensorflow</artifactId>
<version>2.4.0</version>
</dependency>
Dukungan GPU
Jika sistem Anda memiliki dukungan GPU , tambahkan dependensi TensorFlow berikut ke file pom.xml
proyek:
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>libtensorflow</artifactId>
<version>2.4.0</version>
</dependency>
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>libtensorflow_jni_gpu</artifactId>
<version>2.4.0</version>
</dependency>
Program contoh
Contoh ini menunjukkan cara membuat proyek Apache Maven dengan TensorFlow. Pertama, tambahkan dependensi TensorFlow ke file pom.xml
proyek:
<project>
<modelVersion>4.0.0</modelVersion>
<groupId>org.myorg</groupId>
<artifactId>hellotensorflow</artifactId>
<version>1.0-SNAPSHOT</version>
<properties>
<exec.mainClass>HelloTensorFlow</exec.mainClass>
<!-- The sample code requires at least JDK 1.7. -->
<!-- The maven compiler plugin defaults to a lower version -->
<maven.compiler.source>1.7</maven.compiler.source>
<maven.compiler.target>1.7</maven.compiler.target>
</properties>
<dependencies>
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>tensorflow</artifactId>
<version>1.14.0</version>
</dependency>
</dependencies>
</project>
Buat file sumber ( src/main/java/HelloTensorFlow.java
):
import org.tensorflow.Graph;
import org.tensorflow.Session;
import org.tensorflow.Tensor;
import org.tensorflow.TensorFlow;
public class HelloTensorFlow {
public static void main(String[] args) throws Exception {
try (Graph g = new Graph()) {
final String value = "Hello from " + TensorFlow.version();
// Construct the computation graph with a single operation, a constant
// named "MyConst" with a value "value".
try (Tensor t = Tensor.create(value.getBytes("UTF-8"))) {
// The Java API doesn't yet include convenience functions for adding operations.
g.opBuilder("Const", "MyConst").setAttr("dtype", t.dataType()).setAttr("value", t).build();
}
// Execute the "MyConst" operation in a Session.
try (Session s = new Session(g);
// Generally, there may be multiple output tensors,
// all of them must be closed to prevent resource leaks.
Tensor output = s.runner().fetch("MyConst").run().get(0)) {
System.out.println(new String(output.bytesValue(), "UTF-8"));
}
}
}
}
Kompilasi dan jalankan:
mvn -q compile exec:java # Use -q to hide logging
Output perintah: Hello from version
TensorFlow dengan JDK
TensorFlow dapat digunakan dengan JDK melalui Java Native Interface (JNI).
Unduh
- Unduh Arsip Jar TensorFlow (JAR): libtensorflow.jar
- Unduh dan ekstrak file Java Native Interface (JNI) untuk sistem operasi dan dukungan prosesor Anda:
Menyusun
Dengan menggunakan file HelloTensorFlow.java
dari contoh sebelumnya , kompilasi program yang menggunakan TensorFlow. Pastikan libtensorflow.jar
dapat diakses oleh classpath
Anda:
javac -cp libtensorflow-2.4.0.jar HelloTensorFlow.java
Lari
Untuk menjalankan program Java TensorFlow, JVM harus mengakses libtensorflow.jar
dan library JNI yang diekstrak.
Linux / macOS
java -cp libtensorflow-2.4.0.jar:. -Djava.library.path=./jni HelloTensorFlow
Windows
java -cp libtensorflow-2.4.0.jar;. -Djava.library.path=jni HelloTensorFlow
Output perintah: Hello from version
Bangun dari sumber
TensorFlow adalah open source. Baca petunjuk untuk membuat Java TensorFlow dan pustaka asli dari kode sumber.