Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge

Installieren Sie TensorFlow für Java

TensorFlow bietet eine Java-API, mit der Sie mit Python erstellte Modelle laden und in einer Java-Anwendung ausführen können.

Nächtliche Libtensorflow-Java-Pakete

Libtensorflow JNI-Pakete werden jede Nacht erstellt und für alle unterstützten Plattformen auf GCS hochgeladen. Sie werden in den libtensorflow-nightly GCS-Bucket hochgeladen und nach Betriebssystem und Erstellungsdatum indiziert.

Unterstützte Plattformen

TensorFlow für Java wird auf folgenden Systemen unterstützt:

  • Ubuntu 16.04 oder höher; 64-Bit, x86
  • macOS 10.12.6 (Sierra) oder höher
  • Windows 7 oder höher; 64-Bit, x86

Informationen zur Verwendung von TensorFlow unter Android finden Sie unter TensorFlow Lite

TensorFlow mit Apache Maven

Um TensorFlow mit Apache Maven zu verwenden , fügen Sie die Abhängigkeit zur pom.xml Datei des Projekts hinzu:

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

GPU-Unterstützung

Wenn Ihr System GPU-Unterstützung bietet, fügen Sie der Datei pom.xml des Projekts die folgenden TensorFlow-Abhängigkeiten hinzu:

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

Beispielprogramm

Dieses Beispiel zeigt, wie Sie mit TensorFlow ein Apache Maven-Projekt erstellen. Fügen Sie zunächst die TensorFlow-Abhängigkeit zur pom.xml Datei des Projekts hinzu:

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

Erstellen Sie die Quelldatei ( 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"));
      }
    }
  }
}

Kompilieren und ausführen:

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

Der Befehl gibt Hello from version : Hello from version

TensorFlow mit dem JDK

TensorFlow kann mit dem JDK über die Java Native Interface (JNI) verwendet werden.

Herunterladen

  1. Laden Sie das TensorFlow Jar Archive (JAR) herunter : libtensorflow.jar
  2. Laden Sie die JNI-Datei (Java Native Interface) herunter und extrahieren Sie sie für die Unterstützung Ihres Betriebssystems und Prozessors:
JNI-Version URL
Linux
Nur Linux-CPU https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-linux-x86_64-2.4.0.tar.gz
Linux-GPU-Unterstützung https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-linux-x86_64-2.4.0.tar.gz
Mac OS
Nur MacOS-CPU https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-darwin-x86_64-2.4.0.tar.gz
Windows
Nur Windows-CPU https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-windows-x86_64-2.4.0.zip
Windows-GPU-Unterstützung https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-windows-x86_64-2.4.0.zip

Kompilieren

Kompilieren Sie mithilfe der Datei HelloTensorFlow.java aus dem vorherigen Beispiel ein Programm, das TensorFlow verwendet. libtensorflow.jar Sie sicher, dass die libtensorflow.jar für Ihren classpath zugänglich ist:

javac -cp libtensorflow-2.4.0.jar HelloTensorFlow.java

Lauf

Um ein TensorFlow-Java-Programm auszuführen, muss die JVM auf libtensorflow.jar und die extrahierte JNI-Bibliothek zugreifen.

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

Der Befehl gibt Hello from version : Hello from version

Aus der Quelle erstellen

TensorFlow ist Open Source. Lesen Sie die Anweisungen zum Erstellen der Java- und nativen Bibliotheken von TensorFlow aus dem Quellcode.