Java 版 TensorFlow

TensorFlow 提供了一个 Java API,该 API 特别适合加载用 Python 创建的模型并在 Java 应用中运行这些模型。

JAVA API 参考文档:

https://tensorflow.google.cn/api_docs/java/reference/org/tensorflow/package-summary

注意:TensorFlow Java API 不在 TensorFlow API 稳定性保障的涵盖范围内。

 

支持的平台

以下系统支持 Java 版 TensorFlow:

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

要安装 Android 版 TensorFlow,请参阅 Android TensorFlow 支持和 TensorFlow Android 相机演示

 

从源代码编译

TensorFlow 是开源系统。请阅读相关说明,了解如何从源代码编译 TensorFlow 的 Java 和原生库。

 

将 TensorFlow 与 Apache Mave 配合使用

要将 TensorFlow 与 Apache Maven 配合使用,请将以下依赖项添加到项目的 pom.xml 文件中:

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

 

GPU 支持

如果系统支持 GPU,请将以下 TensorFlow 依赖项添加到项目的 pom.xml 文件中:

<dependency>
      <groupId>org.tensorflow</groupId>
      <artifactId>libtensorflow</artifactId>
      <version>1.12.0</version>
    </dependency>
    <dependency>
      <groupId>org.tensorflow</groupId>
      <artifactId>libtensorflow_jni_gpu</artifactId>
      <version>1.12.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.12.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

成功:Java 版 TensorFlow 已配置完毕。

 

将 TensorFlow 与 JDK 配合使用

TensorFlow 可通过 Java 原生接口 (JNI) 与 JDK 配合使用。(以下都是PC端演示)

下载

  1. 下载 TensorFlow Jar Archive (JAR):libtensorflow.jar
  2. 根据您的操作系统和处理器支持下载并解压缩 Java 原生接口 (JNI) 文件:
JNI 版本网址
Linux
Linux(仅支持 CPU)https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-linux-x86_64-1.12.0.tar.gz
Linux(支持 GPU)https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-linux-x86_64-1.12.0.tar.gz
macOS
macOS(仅支持 CPU)https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-darwin-x86_64-1.12.0.tar.gz
Windows
Windows(仅支持 CPU)https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-windows-x86_64-1.12.0.zip
Windows(支持 GPU)https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-windows-x86_64-1.12.0.zip

注意:在 Windows 上,原生库 (tensorflow_jni.dll) 在运行时需要使用 msvcp140.dll。要安装 Visual C++ 2015 Redistributable,请参阅在 Windows 中从源代码编译指南。

编译

使用上一个示例中的 HelloTensorFlow.java 文件,编译使用 TensorFlow 的程序。确保您的 classpath 可以访问 libtensorflow.jar

    javac -cp libtensorflow-1.12.0.jar HelloTensorFlow.java

运行

要执行 TensorFlow Java 程序,JVM 必须访问 libtensorflow.jar 和解压缩的 JNI 库。

Linux / mac

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

Windows

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

上述命令会输出:Hello from version

  • 1
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值