【环境配置】JAVA调用Tensorflow模型

最新推荐文章于 2025-06-19 09:37:54 发布

o(*￣︶￣*)o__小肉松

最新推荐文章于 2025-06-19 09:37:54 发布

阅读量2.9k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习环境配置

本文链接：https://blog.csdn.net/made_in_china_too/article/details/99545146

深度学习同时被 2 个专栏收录

18 篇文章

订阅专栏

环境配置

7 篇文章

订阅专栏

本文详细介绍了在Windows环境下使用Eclipse IDE搭建TensorFlow Java开发环境的过程，并提供了测试代码示例，确保环境配置正确。同时，针对可能出现的错误，如无法找到TensorFlow本地库等问题，给出了具体解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 测试通过的环境

名称	版本
Windows	win10_64位
IDE	Eclipse 2018-12
Tensorflow	1.6.0
JDK	1.8 （Eclipse 2018-12自带）

2. JAVA版本的Tensorflow测试代码

测试代码项目文件百度网盘下载传送门

import org.tensorflow.Graph;
import org.tensorflow.Session;
import org.tensorflow.Tensor;
import org.tensorflow.TensorFlow;


public class test {
	  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"));
		      }
		    }

		  }
}

/* 输出
Hello from 1.6.0
*/

3. 可能遇到的问题及解决方案

3.1 问题：

Exception in thread "main" java.lang.UnsatisfiedLinkError: Cannot find TensorFlow native library for OS: windows, architecture: x86. See https://github.com/tensorflow/tensorflow/tree/master/java/README.md for possible solutions (such as building the library from source).
at org.tensorflow.NativeLibrary.load(NativeLibrary.java:66)
at org.tensorflow.TensorFlow.init(TensorFlow.java:27)
at org.tensorflow.TensorFlow.<clinit>(TensorFlow.java:31)
at ai.advance.project.App.main(App.java:12)

3.2 解决办法：

确保所使用的JDK是1.7版本以上的64位（与电脑位数相一致）
tensorflow_jni.dll要放在项目根目录下。其中tensorflow_jni.dll可以通过解压libtensorflow_jni-1.6.0.jar得到
查看编译器compiler是否支持1.8以上的，我当时使用的MyEclipse2014的compiler最高仅支持1.7。查看compiler的方法如下图（window-preferences，然后直接搜索compiler）