目前 AI 已经火热的不行, 这两天看了看 TensorFlow, 官方对 TensorFlow 的叙述如下:
TensorFlow 是一个使用数据流图进行数值计算的开放源代码软件库. 图中的节点代表数学运算, 而图中的边则代表在这些节点之间传递的多维数组 (张量). 借助这种灵活的架构, 您可以通过一个 API 将计算工作部署到桌面设备, 服务器或移动设备中的一个或多个 CPU 或 GPU.
TensorFlow 官网 https://www.tensorflow.org/
官网的环境中并没有说明我们在 eclipse+JDK 的环境怎么搭建 TensorFlow 的环境, 自己尝试了一下还是比较容易的
TensorFlow JDK+eclipse 环境
1. 下载 libtensorflow.jar https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-1.6.0.jar , 这是 TensorFlow Java 归档 (JAR).
2. 下载 Windows 上适用于 Java 的 TensorFlow https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow_jni-cpu-windows-x86_64-1.6.0.zip 的 Java 原生接口 (JNI) 文件.
3. 解压 zip 文件提取这个里面的. dll 文件.
4. 在 eclipse 中创建一个 Java 工程, 我这直接使用官网给的例子, 代码如下
- public class HelloTF {
- 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); Tensor output = s.runner().fetch("MyConst").run().get(0)) {
- System.out.println(new String(output.bytesValue(), "UTF-8"));
- }
- }
- }
- }
5. Java 工程添加 libtensorflow.jar, 如果不添加这个 jar, 程序会找不到相关的类
6. 最后一步也是, 关键的一步, 我们要加载解压提取的. dll 文件, 怎么做呢, 第一步, 将 tensorflow_jni.dll 文件复制到工程的 src 目录下, 第二步, 在工程上右键属性 ---Java Build Path--->Source, 点击 Source 左边的箭头选择 native library, 点击右边的 edit, 选择项目的 src 目录即可.
这样我们的环境就搭建完成, 最后运行一下项目, 输出 Hello from 版本号, 表示我们的环境已经搭建好了 Hello from 1.6.0
其他语言的环境可以参考官方网站 https://www.tensorflow.org/install/
来源: https://juejin.im/post/5afecad46fb9a07acc11f818