如何在Windows上使用GPU支持编译Tensorflow Java API?

时间:2017-08-16 10:36:31

标签: java windows tensorflow gpu

我在Windows上运行Tensorflow Java API构建了一些模型。但是,我无法在GPU上运行这些模型。

我正在尝试按照here描述的步骤在Windows上编译Tensorflow Java API。但它没有GPU支持或任何补充说明。

请帮助解决这个问题?

2 个答案:

答案 0 :(得分:2)

您好,我不知道问题是否仍然存在,但是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

答案 1 :(得分:0)

目前还不支持Windows + Java + GPU。这是一个相关的github问题https://github.com/tensorflow/tensorflow/issues/10708