TensorFlow는 Python으로 만든 모델을 로드하고 자바 애플리케이션 내에서 실행하는 데 유용한 Java API를 제공합니다.
Nightly Libtensorflow 자바 패키지
Libtensorflow JNI 패키지는 나이틀리로 빌드되며 지원되는 모든 플랫폼의 GCS에 업로드됩니다. libtensorflow-nightly GCS 버킷에 업로드되며 운영체제 및 빌드 날짜에 따라 색인이 생성됩니다.
지원되는 플랫폼
TensorFlow for Java는 다음 시스템에서 지원됩니다.
- Ubuntu 16.04 이상; 64비트, x86
- macOS 10.12.6(Sierra) 이상
- Windows 7 이상, 64비트, x86
Android에서 TensorFlow를 사용하려면 TensorFlow Lite를 참고하세요.
TensorFlow 및 Apache Maven
Apache Maven에서 TensorFlow를 사용하려면 프로젝트의 pom.xml
파일에 종속성을 추가합니다.
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>tensorflow</artifactId>
<version>2.4.0</version>
</dependency>
GPU 지원
시스템에서 GPU를 지원하는 경우 프로젝트의 pom.xml
파일에 다음 TensorFlow 종속성을 추가합니다.
<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>
예제 프로그램
이 예제에서는 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.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
JNI(Java Native Interface)를 통해 JDK에서 TensorFlow를 사용할 수 있습니다.
다운로드
- TensorFlow Jar 보관 파일(JAR) libtensorflow.jar을 다운로드합니다.
- 운영체제 및 프로세서를 지원하는 JNI(Java Native Interface) 파일을 다운로드하여 추출합니다.
컴파일
이전 예제의 HelloTensorFlow.java
파일을 이용하여 TensorFlow를 사용하는 프로그램을 컴파일합니다. libtensorflow.jar
에서 classpath
에 액세스할 수 있는지 확인합니다.
javac -cp libtensorflow-2.4.0.jar HelloTensorFlow.java
실행
TensorFlow 자바 프로그램을 실행하려면 JVM이 libtensorflow.jar
및 추출된 JNI 라이브러리에 액세스해야 합니다.
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
명령어 결과는 Hello from version
입니다.
소스에서 빌드
TensorFlow는 오픈소스입니다. 안내에 따라 소스 코드에서 TensorFlow의 자바 및 네이티브 라이브러리를 빌드합니다.