TensorFlow Java 可在任何 JVM 上运行,用于构建、训练和部署机器学习模型。它既支持 CPU 执行也支持 GPU 执行(无论是图表模式下还是 eager 模式下),并为在 JVM 环境中使用 TensorFlow 提供了丰富的 API。世界各地的大小企业都在频繁使用 Java 和其他 JVM 语言(比如 Scala 和 Kotlin),有鉴于此,若要大规模采用机器学习,TensorFlow Java 无疑是一项战略选择。
注意:TensorFlow Java API 不在 TensorFlow API 稳定性保障的涵盖范围内。
要求
TensorFlow Java 可在 Java 8 及更高版本上运行,并可在以下平台上即开即用:
Ubuntu 16.04 或更高版本;64 位,x86
macOS 10.12.6 (Sierra) 或更高版本;64 位,x86
Windows 7 或更高版本;64 位,x86
注意:若要使用 Android 版 TensorFlow,请参阅 TensorFlow Lite*
版本
TensorFlow Java 有自己独立的发布周期,与 TensorFlow 运行时无关。因此,它的版本与作为其运行环境的 TensorFlow 运行时的版本不匹配。请查阅 TensorFlow Java 版本控制表,了解所有可用的版本以及它们与 TensorFlow 运行时的映射关系。
工件
您可通过多种方式将 TensorFlow Java 添加到您的项目中。最简便的方式是,添加 tensorflow-core-platform 工件的某个依赖项,该工件既包含 TensorFlow J