安装 TensorFlow for Java

TensorFlow 提供 Java API，可用于加载使用 Python 创建的模型并在 Java 应用程序中运行它们。

夜间 Libtensorflow Java 软件包

Libtensorflow JNI 软件包每天晚上构建并上传到 GCS，用于所有支持的平台。它们被上传到 libtensorflow-nightly GCS 存储桶 并按操作系统和构建日期索引。

支持的平台

TensorFlow for Java 在以下系统上受支持

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

要在 Android 上使用 TensorFlow，请参阅 TensorFlow Lite

使用 Apache Maven 的 TensorFlow

要使用 TensorFlow 与 Apache Maven，请将依赖项添加到项目的 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 本地接口 (JNI) 与 JDK 一起使用。

下载

1. 下载 TensorFlow Jar 归档文件 (JAR)：libtensorflow.jar
2. 下载并解压缩适合您的操作系统和处理器支持的 Java 本地接口 (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 和本机库。