[英]How to Compile Tensorflow Java API with GPU Support on Windows?
[英]Tensorflow Java API in windows
我試圖在Windows中為java配置Tensorflow API。
按照我的說法https://github.com/tensorflow/tensorflow/blob/master/tensorflow/java/README.md
它說我們必須構建本地庫,需要從windows的源代碼構建。 但它沒有詳細的說明。 任何人都有運氣得到編譯? 我可以遵循任何步驟嗎?
來自官方網站 :
我們不正式支持在Windows上構建TensorFlow; 但是,如果您不介意在Windows或TensorFlow CMake版本上使用高度實驗性的Bazel,您可以嘗試在Windows上構建TensorFlow。
另外,一個相關的github問題: https : //github.com/tensorflow/tensorflow/issues/17 ,其中提到了一個用戶rongjiecomputer (2016年9月):
Tensorflow要求Bazel從源代碼構建,如果我沒有記錯,Bazel是一個像GNU Make這樣的構建系統,但不是編譯器。
根據我所看到的簡要掃描代碼,Tensorflow本身的源代碼主要使用標准C ++庫來處理線程等問題,因此在Windows上編譯應該沒問題,它使用的基本第三方庫都有Windows支持。
因此,我認為主要問題在於缺乏Windows的構建方法而不是源代碼本身。 有一些使用CMake來構建而不是Bazel但還沒有完成的工作。 如果有人可以將Bazel構建規則轉換為CMake,我認為我們將能夠在Windows上構建它。
所以我假設,如果Bazel Windows正常工作,你可以按照與Bazel在其他平台上構建它的步驟相同。 我沒有嘗試過我自己。
有關將在下一版TensorFlow中用於構建Windows發行二進制文件的腳本,請參閱https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/ci_build/windows/libtensorflow_cpu.sh 。
這顯示了如何使用bazel
(和一些hacks)來生成tensorflow_jni.dll
。
更新 : TensorFlow for Java maven包現在支持Windows,從1.1版開始,還提供了tensorflow_jni.dll
二進制文件。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.