我正在尝试在Tensorflow中构建一个在急切执行模式下的图像分类器。我收到以下错误。在
InternalError: Could not find valid device for node.
Node: {{node Conv2D}} = Conv2D[T=DT_UINT8, data_format="NHWC",
dilations=[1, 1, 1, 1], padding="SAME", strides=[1, 2, 2, 1],
use_cudnn_on_gpu=true](dummy_input, dummy_input)
All kernels registered for op Conv2D :
device='XLA_CPU'; T in [DT_FLOAT, DT_DOUBLE, DT_HALF]
device='XLA_GPU'; T in [DT_FLOAT, DT_DOUBLE, DT_BFLOAT16,
DT_HALF]
device='GPU'; T in [DT_DOUBLE]
device='GPU'; T in [DT_FLOAT]
device='GPU'; T in [DT_HALF]
device='CPU'; T in [DT_DOUBLE]
device='CPU'; T in [DT_FLOAT]
device='CPU'; T in [DT_HALF]
device='XLA_CPU_JIT'; T in [DT_FLOAT, DT_DOUBLE, DT_HALF]
device='XLA_GPU_JIT'; T in [DT_FLOAT, DT_DOUBLE, DT_BFLOAT16,
DT_HALF]
[Op:Conv2D]
有什么想法为什么?在
错误是因为您试图将int32张量加载到conv2d块。在将张量传递给特定的tensorflow函数之前,必须检查支持的输入张量类型。在
例如,如果您签出conv2d的tensorflow源代码tf.nn.conv2d,那么支持的输入类型是
half
,bfloat16
,float32
,float64
。在这也会在您的错误语句后指出-“所有内核注册为op Conv2D”
相关问题 更多 >
编程相关推荐