![](/img/trans.png)
[英]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.