Tensorflow为什么不保存run_metadata?

时间:2018-08-13 22:05:11

标签: python-3.x tensorflow machine-learning tensorboard

我只是试图生成一个摘要,以显示run_metadata,如下所示:

run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
run_metadata = tf.RunMetadata()
summary = sess.run([x, y], options=run_options, run_metadata=run_metadata)

train_writer.add_run_metadata(paths.logs, 'step%d' % step)
train_writer.add_summary(paths.logs, step)

我确保了logs文件夹的路径存在,这可以通过以下事实得到确认:生成了摘要文件,但未预置任何元数据。现在我不确定是否真的生成了一个文件(对于元数据而言),但是当我打开张量板时,图形看起来很好,并且填充了session runs下拉菜单。现在,当我选择任何运行时,它都会显示一个进度条“ Parsing metadata.pbtxt”,它会停止并在一半时间内挂起。

这使我无法收集有关图形的更多其他信息。我想念什么吗?尝试在本地运行本教程(MNIST summary tutorial)时发生了类似的问题。我觉得我缺少一些简单的东西。有谁知道可能导致此问题的原因吗?尝试加载会话运行数据时,为什么我的张量板会挂起?

1 个答案:

答案 0 :(得分:1)

我无法相信我在发布问题后立即使它起作用,但问题就解决了。我注意到这一行:

run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)

给我一​​个错误,所以我删除了参数并将其变成

run_options = tf.RunOptions()

没有意识到这就是导致元数据无法解析的原因。研究错误消息后:

Couldn't open CUDA library cupti64_90.dll

我查看了这个Github Thread,并将文件移到bin文件夹中。之后,我再次使用trace_level参数运行我的代码,没有错误,并且元数据已成功解析。