ติดตั้ง TensorFlow สำหรับ Java

TensorFlow มี API สำหรับภาษา Java ซึ่งมีประโยชน์สำหรับการโหลดโมเดลที่สร้างด้วย Python และเรียกใช้งานภายในแอปพลิเคชัน Java

แพ็กเกจ Java Libtensorflow เวอร์ชันกลางคืน

แพ็กเกจ JNI ของ Libtensorflow จะถูกสร้างขึ้นทุกคืนและอัปโหลดไปยัง GCS สำหรับทุกแพลตฟอร์มที่รองรับ โดยจะถูกอัปโหลดไปยัง บัคเก็ต GCS ชื่อ libtensorflow-nightly และมีการจัดทำดัชนีตามระบบปฏิบัติการและวันที่สร้าง

แพลตฟอร์มที่รองรับ

TensorFlow สำหรับ Java รองรับการทำงานบนระบบต่อไปนี้:

  • Ubuntu 16.04 หรือสูงกว่า; 64 บิต, x86
  • macOS 10.12.6 (Sierra) หรือสูงกว่า
  • ระบบปฏิบัติการ Windows 7 หรือสูงกว่า; 64 บิต, x86

หากต้องการใช้งาน TensorFlow บน Android โปรดดู TensorFlow Lite

TensorFlow กับ Apache Maven

ในการใช้งาน TensorFlow ร่วมกับ Apache Maven ให้เพิ่ม dependency ลงในไฟล์ pom.xml ของโปรเจ็กต์:

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

การรองรับ GPU

หากระบบของคุณ รองรับ GPU ให้เพิ่ม 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 ขั้นแรก ให้เพิ่ม dependency ของ 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 Native Interface (JNI)

ดาวน์โหลด

  1. ดาวน์โหลดไฟล์ JAR ของ TensorFlow: libtensorflow.jar
  2. ดาวน์โหลดและแตกไฟล์ Java Native Interface (JNI) ที่เหมาะสมกับระบบปฏิบัติการและโปรเซสเซอร์ของคุณ:
เวอร์ชัน JNI URL
ลินุกซ์
ซีพียู Linux เท่านั้น https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-linux-x86_64-2.4.0.tar.gz
การรองรับ GPU ของ Linux https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-linux-x86_64-2.4.0.tar.gz
ระบบปฏิบัติการ macOS
เฉพาะ CPU ของ macOS เท่านั้น https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-darwin-x86_64-2.4.0.tar.gz
วินโดวส์
เฉพาะ CPU ของ Windows เท่านั้น https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-windows-x86_64-2.4.0.zip
การรองรับ GPU ของ 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

วิ่ง

ในการเรียกใช้โปรแกรม 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 เป็นโอเพนซอร์ส โปรดอ่าน คำแนะนำ ในการสร้างไลบรารี Java และไลบรารีเนทีฟของ TensorFlow จากซอร์สโค้ด