我已经用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训练的模型"环境?