簡體   English   中英

Tensorflow 訓練錯誤 model(在 GPU 上)

[英]Tensorflow error upon training model (on GPU)

我正在使用Tanner Gilbert Tensorflow 教程,到目前為止,我已經解決了所有錯誤,讓我進入了實際訓練的最后階段,在第 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

在他的視頻教程中,他說您可以根據 gpu 強度使用 batch_size 值 4,8 等。 但是他在屏幕上的代碼中輸入了 1,我從 4 變成了 1,現在訓練開始了

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM