TensorFlow предоставляет API Java , полезный для загрузки моделей, созданных с помощью Python, и запуска их в приложении Java.
Ночные Java-пакеты Libtensorflow
Пакеты JNI Libtensorflow собираются каждую ночь и загружаются в GCS для всех поддерживаемых платформ. Они загружаются в корзину GCS libtensorflow-nightly и индексируются по операционной системе и дате создания.
Поддерживаемые платформы
TensorFlow для Java поддерживается в следующих системах:
- Ubuntu 16.04 или выше; 64-битная, x86
- macOS 10.12.6 (Сьерра) или новее
- Windows 7 или выше; 64-битная, x86
Чтобы использовать TensorFlow на Android, см. TensorFlow Lite.
TensorFlow с Apache Maven
Чтобы использовать TensorFlow с Apache Maven , добавьте зависимость в файл pom.xml
проекта:
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>tensorflow</artifactId>
<version>2.4.0</version>
</dependency>
Поддержка графического процессора
Если ваша система поддерживает графический процессор , добавьте следующие зависимости 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>
Пример программы
В этом примере показано, как создать проект Apache Maven с помощью TensorFlow. Сначала добавьте зависимость 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
TensorFlow с JDK
TensorFlow можно использовать с JDK через собственный интерфейс Java (JNI).
Скачать
- Загрузите архив TensorFlow Jar (JAR): libtensorflow.jar .
- Загрузите и извлеките файл Java Native Interface (JNI) для вашей операционной системы и поддержки процессора:
JNI-версия | URL-адрес |
---|---|
Линукс | |
Только процессор Linux | https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-linux-x86_64-2.4.0.tar.gz |
Поддержка графического процессора Linux | https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-linux-x86_64-2.4.0.tar.gz |
macOS | |
только процессор MacOS | https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-darwin-x86_64-2.4.0.tar.gz |
Окна | |
Только процессор Windows | https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-windows-x86_64-2.4.0.zip |
Поддержка графического процессора Windows | https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-windows-x86_64-2.4.0.zip |
Скомпилировать
Используя файл HelloTensorFlow.java
из предыдущего примера , скомпилируйте программу, использующую TensorFlow. Убедитесь, что libtensorflow.jar
доступен для вашего classpath
:
javac -cp libtensorflow-2.4.0.jar HelloTensorFlow.java
Бегать
Чтобы выполнить Java-программу TensorFlow, 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 имеет открытый исходный код. Прочтите инструкции по сборке Java и собственных библиотек TensorFlow из исходного кода.