训练模型上的 TensorFlow 错误（在 GPU 上）

Posted 2023-02-16

【中文标题】训练模型上的 TensorFlow 错误（在 GPU 上）

【英文标题】：Tensorflow error upon training model (on GPU)

【发布时间】：2021-04-19 13:20:38

【问题描述】：

我正在使用Tanner Gilbert Tensorflow tutorial，到目前为止，我已经解决了所有错误，以使我进入实际训练的最后阶段，在第 6 步，我收到此错误，这是错误消息的最后一部分，任何人都可以修复?

tensorflow.python.framework.errors_impl.NotFoundError: Could not find valid device for node.
Node:node HashTableV2
All kernels registered for op HashTableV2 :
  device='CPU'; key_dtype in [DT_INT32]; value_dtype in [DT_DOUBLE]
  device='CPU'; key_dtype in [DT_INT32]; value_dtype in [DT_FLOAT]
  device='CPU'; key_dtype in [DT_INT32]; value_dtype in [DT_INT32]
  device='CPU'; key_dtype in [DT_INT32]; value_dtype in [DT_STRING]
  device='CPU'; key_dtype in [DT_INT64]; value_dtype in [DT_DOUBLE]
  device='CPU'; key_dtype in [DT_INT64]; value_dtype in [DT_FLOAT]
  device='CPU'; key_dtype in [DT_INT64]; value_dtype in [DT_INT32]
  device='CPU'; key_dtype in [DT_INT64]; value_dtype in [DT_INT64]
  device='CPU'; key_dtype in [DT_INT64]; value_dtype in [DT_STRING]
  device='CPU'; key_dtype in [DT_STRING]; value_dtype in [DT_BOOL]
  device='CPU'; key_dtype in [DT_STRING]; value_dtype in [DT_DOUBLE]
  device='CPU'; key_dtype in [DT_STRING]; value_dtype in [DT_FLOAT]
  device='CPU'; key_dtype in [DT_STRING]; value_dtype in [DT_INT32]
  device='CPU'; key_dtype in [DT_STRING]; value_dtype in [DT_INT64]
  device='CPU'; key_dtype in [DT_STRING]; value_dtype in [DT_STRING]
 [Op:HashTableV2] name: hash_table

【参考方案1】：

在他的视频教程中，他说您可以根据 gpu 强度使用 batch_size 值 4,8 等。但是他在屏幕上的代码中输入了 1，我从 4 更改为 1，现在正在训练中