tensorflow Tensor2Tensor中Tensor打印形状

9nvpjoqh  于 2022-11-16  发布在  其他
关注(0)|答案(1)|浏览(192)

我尝试使用tensor 2 tensor打印Tensor的形状(运行基本转换器代码)。我看到形状中有“?”。我尝试执行以下操作:

#with tf.Session() as sess:
    #  print(sess.run(x).shape)

但这给了我一个错误

cannot assign a device for operation transformer/parallel_0/transformer/Identity: {{node transformer/parallel_0/transformer/Identity}} was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0 ]. Make sure the device specification refers to a valid device.
         [[transformer/parallel_0/transformer/Identity]]

我也尝试过使用tf.print(tf.shape(x)),但它什么也打印不出来
我正在运行命令:

t2t-decoder \
  --data_dir=$DATA_DIR \
  --problem=$PROBLEM \
  --model=$MODEL \
  --hparams_set=$HPARAMS \
  --output_dir=$TRAIN_DIR \
  --decode_hparams="beam_size=$BEAM_SIZE,alpha=$ALPHA" \
  --decode_from_file=$DECODE_FILE \
  --decode_to_file=translation.en

我不太了解tensorflow ,但我看到这些正在使用?

tf.compat.v1.train.MonitoredSession(
            session_creator=tf.compat.v1.train.ChiefSessionCreator(

在这种情况下,我如何打印Tensor的形状?请帮助!!!

58wvjzkj

58wvjzkj1#

无法为操作transformer/parallel_0/transformer/Identity分配设备:{{node transformer/parallel_0/transformer/Identity}}已明确指派给/device:GPU:0,但可用的装置为[ /job:localhost/replica:0/task:0/device:CPU:0 ]。请确定装置规格指涉有效的装置。transformer/parallel_0/transformer/Identity
这个错误告诉您您的程式码尝试在GPU上执行,但是找不到或无法存取它。您安装了tensorflowtensorflow-gpu吗?如果您要使用GPU,就需要安装后者。
如果您已经安装了tensorflow-gpu并且错误仍然存在,那么您可能需要检查您的tensroflow/cuda/python版本兼容性。Here is how you can do that.

相关问题