TensorFlow Java+eclipse下環境搭建

maweiliang發表於2018-05-18

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

TensorFlow是一個使用資料流圖進行數值計算的開放原始碼軟體庫。圖中的節點代表數學運算,而圖中的邊則代表在這些節點之間傳遞的多維陣列(張量)。藉助這種靈活的架構,您可以通過一個 API 將計算工作部署到桌面裝置、伺服器或移動裝置中的一個或多個 CPU 或 GPU。

TensorFlow官網

官網的環境中並沒有說明我們在eclipse+JDK的環境怎麼搭建TensorFlow的環境,自己嘗試了一下還是比較容易的

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
複製程式碼

其他語言的環境可以參考官方網站

相關文章