【问题标题】:Tensorflow error upon training model (on GPU)训练模型上的 TensorFlow 错误(在 GPU 上)
【发布时间】:2023-09-18 00:14:02
【问题描述】:

我正在使用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

【问题讨论】:

    标签: tensorflow machine-learning deep-learning gpu


    【解决方案1】:

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

    【讨论】:

      最近更新 更多