一个 TF 图可以在一个会话中不离开 GPU 的情况下输入另一个图吗?
Can a TF graph feed into another without leaving the GPU in one session?
假设我有两个 TF 图 G_1
、G_2
,我想获取 G_1
的输出并将其输入 G_2
。我的假设是,要做到这一点,我们在 G_1
的馈送下进入 GPU,然后获取 G_1
的输出并将其馈送到 G_2
输出到 CPU。
有什么方法可以在 GPU 上提供两个图形吗?
In general you do not have to specify CPUs or GPUs explicitly. TensorFlow uses your first GPU, if you have one, for as many operations as possible.
G_1
的输出将进入 G-2
,它也将在 GPU 上执行。
假设我有两个 TF 图 G_1
、G_2
,我想获取 G_1
的输出并将其输入 G_2
。我的假设是,要做到这一点,我们在 G_1
的馈送下进入 GPU,然后获取 G_1
的输出并将其馈送到 G_2
输出到 CPU。
有什么方法可以在 GPU 上提供两个图形吗?
In general you do not have to specify CPUs or GPUs explicitly. TensorFlow uses your first GPU, if you have one, for as many operations as possible.
G_1
的输出将进入 G-2
,它也将在 GPU 上执行。