如何在 Windows 上编译具有 GPU 支持的 Tensorflow Java API？
[英]How to Compile Tensorflow Java API with GPU Support on Windows?
问题描述
我有一些使用在 Windows 上运行的 Tensorflow Java API 构建的模型。 但是，我无法在 GPU 上运行这些模型。
我正在尝试按照此处描述的步骤在 Windows 上编译 Tensorflow Java API。 但它没有 GPU 支持或任何补充说明。
请问能帮忙解决这个问题吗？
2 个解决方案
解决方案1
2 已采纳 2020-02-05 14:08:34
您好，我不知道这个问题是否仍然相关，但 Tensorflow 已经为 Windows 提供了 GPU 支持。 只需确保为您的 Java TF 版本安装了正确版本的 CUDA，并且您的 GPU 具有最低计算能力。
CUDA 版本可能有点棘手。 我尝试了 tf 1.12 和 tf 1.15。 tf 1.12 适用于 CUDA 9.0，tf 1.15 适用于 CUDA 10.0
解决方案2
0 2017-08-23 23:39:49
尚不支持 Windows+Java+GPU。 这是一个相关的 github 问题https://github.com/tensorflow/tensorflow/issues/10708
如何确保我的 tensorFlow Java 程序在 Windows 上使用我的 GPU？
[英]How can I make sure that my tensorFlow Java program is using my GPU on Windows?
通过Java API在Windows上使用Tensorflow.contrb中的操作
[英]Using ops from Tensorflow.contrb on Windows via Java API
声明:本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.
相关问题
如何在 GPU 支持下编译 tflite？
Windows中的Tensorflow Java API
如何确保我的 tensorFlow Java 程序在 Windows 上使用我的 GPU？
在GPU Java上配置Tensorflow
Windows 对 Java 6 中原生 GSS-API 的支持
如何编译支持旧版 Java 的 .java？
Tensorflow Java多GPU推理
如何编译Windows版Open Kinect以用于Java？
通过Java API在Windows上使用Tensorflow.contrb中的操作
如何在Tensorflow Java API中设置会话配置？
相关标签