TensorFlow را برای جاوا نصب کنید

TensorFlow یک API جاوا ارائه می‌دهد که برای بارگذاری مدل‌های ایجاد شده با پایتون و اجرای آنها در یک برنامه جاوا مفید است.

بسته‌های جاوای شبانه‌ی Libtensorflow

بسته‌های JNI مربوط به Libtensorflow به صورت شبانه ساخته می‌شوند و برای همه پلتفرم‌های پشتیبانی‌شده در GCS آپلود می‌شوند. آن‌ها در باکت GCS مربوط به libtensorflow-nightly آپلود می‌شوند و بر اساس سیستم عامل و تاریخ ساخت، فهرست‌بندی می‌شوند.

پلتفرم‌های پشتیبانی‌شده

TensorFlow برای جاوا در سیستم‌های زیر پشتیبانی می‌شود:

  • اوبونتو ۱۶.۰۴ یا بالاتر؛ ۶۴ بیتی، x86
  • macOS 10.12.6 (سیرا) یا بالاتر
  • ویندوز ۷ یا بالاتر؛ ۶۴ بیتی، x86

برای استفاده از TensorFlow در اندروید، به TensorFlow Lite مراجعه کنید.

تنسورفلو با آپاچی ماون

برای استفاده از TensorFlow با Apache Maven ، وابستگی را به فایل pom.xml پروژه اضافه کنید:

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

پشتیبانی از پردازنده گرافیکی

اگر سیستم شما از 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>

برنامه نمونه

این مثال نحوه ساخت یک پروژه 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

تنسورفلو با JDK

TensorFlow را می‌توان از طریق رابط بومی جاوا (JNI) با JDK استفاده کرد.

دانلود

  1. دانلود فایل فشرده TensorFlow Jar (JAR): libtensorflow.jar
  2. فایل رابط بومی جاوا (JNI) را متناسب با سیستم عامل و پردازنده خود دانلود و استخراج کنید:
نسخه JNI آدرس اینترنتی
لینوکس
فقط پردازنده لینوکس https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-linux-x86_64-2.4.0.tar.gz
پشتیبانی از پردازنده گرافیکی لینوکس https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-linux-x86_64-2.4.0.tar.gz
مک‌او‌اس
فقط پردازنده macOS https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-darwin-x86_64-2.4.0.tar.gz
ویندوز
فقط پردازنده ویندوز https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-windows-x86_64-2.4.0.zip
پشتیبانی از پردازنده گرافیکی ویندوز 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

اجرا کنید

برای اجرای یک برنامه جاوا 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 متن‌باز است. دستورالعمل‌های ساخت کتابخانه‌های جاوا و بومی TensorFlow از کد منبع را بخوانید.