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 استفاده کرد.
دانلود
- دانلود فایل فشرده TensorFlow Jar (JAR): libtensorflow.jar
- فایل رابط بومی جاوا (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 از کد منبع را بخوانید.