TensorFlow Java+eclipse下環境搭建

目前AI已經火熱的不行,這兩天看了看TensorFlow,官方對TensorFlow的敘述以下:windows

TensorFlow是一個使用數據流圖進行數值計算的開放源代碼軟件庫。圖中的節點表明數學運算,而圖中的邊則表明在這些節點之間傳遞的多維數組(張量)。藉助這種靈活的架構,您能夠經過一個 API 將計算工做部署到桌面設備、服務器或移動設備中的一個或多個 CPU 或 GPU。api

TensorFlow官網數組

官網的環境中並無說明咱們在eclipse+JDK的環境怎麼搭建TensorFlow的環境,本身嘗試了一下仍是比較容易的bash

TensorFlow JDK+eclipse環境

1.下載 libtensorflow.jar,這是 TensorFlow Java 歸檔 (JAR)。
2.下載 Windows 上適用於 Java 的 TensorFlow 的 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
複製代碼

其餘語言的環境能夠參考官方網站服務器

相關文章
相關標籤/搜索