Tensorflow ERROR：tf.one_hot执行失败 Could not find valid device for node

错误：
NotFoundError: Could not find valid device for node.
Node:{{node OneHot}}
All kernels registered for op OneHot :
device=‘GPU’; TI in [DT_UINT8]; T in [DT_HALF]
device=‘GPU’; TI in [DT_INT32]; T in [DT_HALF]
.
.
.
device=‘CPU’; TI in [DT_INT64]; T in [DT_VARIANT]
[Op:OneHot] name: one_hot/

这句一直报错：

y_ = tf.one_hot(y_train, depth=2)

然后发现y_train没有进行转化成int型，在这前面加一句：

y_train = tf.cast(y_train, tf.int32)

解决。