我尝试使用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的形状?请帮助!!!
1条答案
按热度按时间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上执行,但是找不到或无法存取它。您安装了
tensorflow
或tensorflow-gpu
吗?如果您要使用GPU,就需要安装后者。如果您已经安装了tensorflow-gpu并且错误仍然存在,那么您可能需要检查您的tensroflow/cuda/python版本兼容性。Here is how you can do that.