TensorFlow for Java 설치

TensorFlow는 Python으로 만든 모델을 로드하고 자바 애플리케이션 내에서 실행하는 데 특히 유용한 자바 API를 제공합니다.

지원되는 플랫폼

TensorFlow for Java는 다음 시스템에서 지원됩니다.

  • Ubuntu 16.04 이상; 64비트, x86
  • macOS 10.12.6(Sierra) 이상
  • Windows 7 이상, 64비트, x86

Android에 TensorFlow를 설치하려면 Android TensorFlow 지원TensorFlow Android 카메라 데모를 참조하세요.

TensorFlow 및 Apache Maven

Apache Maven에서 TensorFlow를 사용하려면 프로젝트의 pom.xml 파일에 종속성을 추가합니다.

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

GPU 지원

시스템에 GPU 지원이 있는 경우 프로젝트의 pom.xml 파일에 다음 TensorFlow 종속성을 추가합니다.

<dependency>
      <groupId>org.tensorflow</groupId>
      <artifactId>libtensorflow</artifactId>
      <version>1.13.1</version>
    </dependency>
    <dependency>
      <groupId>org.tensorflow</groupId>
      <artifactId>libtensorflow_jni_gpu</artifactId>
      <version>1.13.1</version>
    </dependency>
    

예제 프로그램

이 예제에서는 TensorFlow를 사용하여 Apache Maven 프로젝트를 빌드하는 방법을 보여줍니다. 먼저 프로젝트의 pom.xml 파일에 TensorFlow 종속성을 추가합니다.

<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.13.1</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

JNI(Java Native Interface)를 통해 JDK에서 TensorFlow를 사용할 수 있습니다.

다운로드

  1. TensorFlow Jar 보관 파일(JAR) libtensorflow.jar을 다운로드합니다.
  2. 운영체제 및 프로세서를 지원하는 JNI(Java Native Interface) 파일을 다운로드하여 추출합니다.
JNI 버전URL
Linux
Linux CPU만 https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-linux-x86_64-1.13.1.tar.gz
Linux GPU 지원 https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-linux-x86_64-1.13.1.tar.gz
macOS
macOS CPU만 https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-darwin-x86_64-1.13.1.tar.gz
Windows
Windows CPU만 https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-windows-x86_64-1.13.1.zip
Windows GPU 지원 https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-gpu-windows-x86_64-1.13.1.zip

컴파일

이전 예제HelloTensorFlow.java 파일을 이용하여 TensorFlow를 사용하는 프로그램을 컴파일합니다. classpath에서 libtensorflow.jar에 액세스할 수 있는지 확인합니다.

    javac -cp libtensorflow-1.13.1.jar HelloTensorFlow.java
    

실행

TensorFlow 자바 프로그램을 실행하려면 JVM이 libtensorflow.jar 및 추출된 JNI 라이브러리에 액세스해야 합니다.

Linux/macOS

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

Windows

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

이 명령어는 다음을 출력합니다. Hello from version

소스에서 빌드

TensorFlow는 오픈 소스입니다. 안내에 따라 소스 코드에서 TensorFlow의 자바 및 네이티브 라이브러리를 빌드합니다.