微调微调是否需要gpu?

时间:2017-06-27 05:53:36

标签: machine-learning tensorflow deep-learning tf-slim

实际上我试图在git hub上使用tf slim微调示例微调inceptionV3模型它给了我这个错误:

InvalidArgumentError(参见上面的回溯):无法将设备分配给节点'InceptionV3 / AuxLogits / Conv2d_2b_1x1 / biases / RMSProp_1':无法满足显式设备规范'/ device:GPU:0',因为没有匹配该规范的设备是在此过程中注册;可用设备:/ job:localhost / replica:0 / task:0 / cpu:0
主机托管调试信息:
主机托管组具有以下类型和设备:
ApplyRMSProp:CPU
Const:CPU
分配:CPU
IsVariableInitialized:CPU
身份:CPU
VariableV2:CPU
     [[节点:InceptionV3 / AuxLogits / Conv2d_2b_1x1 / biases / RMSProp_1 = VariableV2_class = [“loc:@ InceptionV3 / AuxLogits / Conv2d_2b_1x1 / biases”],container =“”,dtype = DT_FLOAT,shape = [5],shared_name =“” ,_device =“/ device:GPU:0”]]

1 个答案:

答案 0 :(得分:0)

请提供有关Tensorflow安装(GPU或CPU安装)的更多信息。如果您正在运行GPU Tensorflow版本,则可能会引发错误。