如何在 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中設置會話配置？
相關標簽