tensorflow使用gpus训练模型,并使用java api进行推理

时间:2017-09-26 15:31:09

标签: spring tensorflow tensorflow-gpu spring-cloud-dataflow

我已经用4个GPU训练了一个张量流模型。我会用它来推断使用tensorflow Java API(在spring cloud dataflow中),但是我得到了这个错误:

  

org.springframework.messaging.MessageHandlingException:嵌套   异常是java.lang.IllegalArgumentException:无法分配   设备到节点&embhers / embedding_matrix / read':无法满足   显式设备规范' /设备:GPU:0'因为没有设备   匹配该规范在此过程中注册;可得到   devices:/ job:localhost / replica:0 / task:0 / cpu:0

我无法理解这个问题。是不是可以在标准中使用经过GPU训练的模型"环境?

0 个答案:

没有答案