我在Windows上运行Tensorflow Java API构建了一些模型。但是,我无法在GPU上运行这些模型。
我正在尝试按照here描述的步骤在Windows上编译Tensorflow Java API。但它没有GPU支持或任何补充说明。
请帮助解决这个问题?
答案 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