How to Compile Tensorflow Java API with GPU Support on Windows?
Question
I have some models build with Tensorflow Java API running on Windows. But, I can't run these models on GPU.
I am trying to compile Tensorflow Java API on Windows following the steps described here . But it does not have GPU support or any complementar instruction for that.
Please, could help to solve this issues?
2 answers
solution1
2 ACCPTED 2020-02-05 14:08:34
Hello I do not know if the issue is still relevant, but Tensorflow already provides GPU support for Windows. Just make sure you have the correct version of CUDA installed for your version of Java TF and that your GPU has the minimum computing capability.
The CUDA versions might be a little tricky. I tried with both tf 1.12 and tf 1.15. Tf 1.12 worked with CUDA 9.0 and tf 1.15 worked with CUDA 10.0
solution2
0 2017-08-23 23:39:49
Windows+Java+GPU is not supported yet. Here is a related github issue https://github.com/tensorflow/tensorflow/issues/10708
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.