tensorflow session 运行 是否与我的其余代码并行?
Is tensorflow session running in parallel to the rest of my code?
我 运行 我在 GPU 上进行会话,我想知道 'session.run()' 代码段是否 运行 与脚本中的其他代码并行。
我在循环中 运行 'session.run()' 之前对 CPU 使用批处理,并希望通过 GPU 上的执行来流水线处理此处理。这个设置已经满足了吗还是我需要手动启动线程?
这完全取决于您编写代码的方式。通过同时检查您的 CPU 和 GPU 利用率
,这应该很容易检查
我通常使用tf.data.Dataset
API。我使用迭代器的 get_next()
方法将数据提供给网络。 CPU 和 GPU 在这种情况下并行工作。
经过一些研究,我发现 'session.run' 不是 运行 同时与您的其他代码。事实上,正如 Ujjwal 所建议的那样,'tf.data.Dataset' API 是流水线批处理预处理和 GPU 执行的最佳选择。
我 运行 我在 GPU 上进行会话,我想知道 'session.run()' 代码段是否 运行 与脚本中的其他代码并行。 我在循环中 运行 'session.run()' 之前对 CPU 使用批处理,并希望通过 GPU 上的执行来流水线处理此处理。这个设置已经满足了吗还是我需要手动启动线程?
这完全取决于您编写代码的方式。通过同时检查您的 CPU 和 GPU 利用率
,这应该很容易检查我通常使用tf.data.Dataset
API。我使用迭代器的 get_next()
方法将数据提供给网络。 CPU 和 GPU 在这种情况下并行工作。
经过一些研究,我发现 'session.run' 不是 运行 同时与您的其他代码。事实上,正如 Ujjwal 所建议的那样,'tf.data.Dataset' API 是流水线批处理预处理和 GPU 执行的最佳选择。