安裝適用於 Java 的 TensorFlow

TensorFlow 提供 Java API，可用於載入以 Python 建立的模型，並在 Java 應用程式中執行。

Libtensorflow Java 每夜建構套件

Libtensorflow JNI 套件為每夜建構，並上傳至 GCS 以供所有支援的平台使用。這些套件會上傳至 libtensorflow-nightly GCS 值區，並依作業系統和建構日期編製索引。

支援的平台

適用於 Java 的 TensorFlow 在下列系統上受到支援

• Ubuntu 16.04 或更高版本；64 位元，x86
• macOS 10.12.6 (Sierra) 或更高版本
• Windows 7 或更高版本；64 位元，x86

若要在 Android 上使用 TensorFlow，請參閱 TensorFlow Lite

搭配 Apache Maven 的 TensorFlow

若要搭配 Apache Maven 使用 TensorFlow，請將依附元件新增至專案的 pom.xml 檔案

<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>tensorflow</artifactId>
  <version>2.4.0</version>
</dependency>

GPU 支援

如果您的系統有 GPU 支援，請將下列 TensorFlow 依附元件新增至專案的 pom.xml 檔案

<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>

範例程式

這個範例示範如何使用 TensorFlow 建構 Apache Maven 專案。首先，將 TensorFlow 依附元件新增至專案的 pom.xml 檔案

<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>

建立原始碼檔案 (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"));
      }
    }
  }
}

編譯並執行

mvn -q compile exec:java  # Use -q to hide logging

指令輸出：Hello from version

搭配 JDK 的 TensorFlow

TensorFlow 可以透過 Java Native Interface (JNI) 搭配 JDK 使用。

下載

1. 下載 TensorFlow Jar 封存檔 (JAR)：libtensorflow.jar
2. 下載並解壓縮適用於您的作業系統和處理器支援的 Java Native Interface (JNI) 檔案

編譯

使用先前範例中的 HelloTensorFlow.java 檔案，編譯使用 TensorFlow 的程式。請確認 libtensorflow.jar 可供您的 classpath 存取

javac -cp libtensorflow-2.4.0.jar HelloTensorFlow.java

執行

若要執行 TensorFlow Java 程式，JVM 必須存取 libtensorflow.jar 和解壓縮的 JNI 程式庫。

java -cp libtensorflow-2.4.0.jar:. -Djava.library.path=./jni HelloTensorFlow

java -cp libtensorflow-2.4.0.jar;. -Djava.library.path=jni HelloTensorFlow

指令輸出：Hello from version

從原始碼建構

TensorFlow 是開放原始碼。請閱讀相關操作說明，以從原始碼建構 TensorFlow 的 Java 和原生程式庫。