tensorflow/java

Java 版 TensorFlow

是什么 – 一组 Java 语言绑定，可让您在任何 JVM 上运行 TensorFlow 模型。该仓库包含三个主要构件：

• tensorflow-core – 对原生 TensorFlow 库的轻量级 JNI 包装。如果您只需要低级别访问，请使用此组件。
• tensorflow-framework – 用于在 Java 中直接构建和训练神经网络的高级 API。
• tensorflow-ndarray – 一个独立的 n 维数组库，可脱离 TensorFlow 其余部分使用。

为何重要 – Java、Scala、Kotlin 等 JVM 语言在大型企业中很常见。通过提供 Maven 风格的构件，TensorFlow-Java 让这些团队能够在不离开现有构建和部署流水线的情况下采用机器学习。

快速入门 – 添加适当的 Maven（或 Gradle）依赖项。您可以选择：

1. 单独选择原生二进制文件（tensorflow-core-api + tensorflow-core-native，并附加平台分类器），以保持 JAR 文件较小；或
2. 使用单一的 tensorflow-core-platform 构件，该构件捆绑了所有支持的 CPU 平台（Linux x86_64、Linux arm64、macOS arm64）的 API 和原生库。

示例（Maven，Linux-x86_64，仅 CPU）：

<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>tensorflow-core-api</artifactId>
  <version>1.2.0</version>
</dependency>
<dependency>
  <groupId>org.tensorflow</groupId>
  <artifactId>tensorflow-core-native</artifactId>
  <version>1.2.0</version>
  <classifier>linux-x86_64</classifier>
</dependency>

如需 GPU 支持，请添加 linux-x86_64-gpu 分类器，并确保已安装 NVIDIA 驱动、CUDA 工具包和 cuDNN。

版本兼容性 – README 中的表格将每个 TensorFlow-Java 版本映射到其底层 TensorFlow 版本和所需的最低 Java 版本（例如，TF-Java 1.2.0 与 TensorFlow 2.21.0 兼容，需要 Java 11）。

社区与贡献 – 由 TensorFlow JVM SIG 维护。您可以在 TensorFlow 论坛（sig_jvm 标签）或 Gitter 频道提问，贡献遵循 CONTRIBUTING.md 中描述的常规拉取请求工作流程。

实际应用示例 – 示例模型位于配套仓库 https://github.com/tensorflow/java-models 中。

---

• 上述所有细节均直接取自仓库的 README。*