TF图可以在不退出GPU的情况下进入另一个会话吗?

时间:2016-05-30 16:51:13

标签: tensorflow

假设我有两个TF图G_1G_2,我想取G_1的输出并将其输入G_2。我的假设是,为此,我们在G_1的Feed上输入GPU,然后取G_1的输出并将其输入G_2,然后输出到CPU。

有没有办法在GPU上提供两个图形?

1 个答案:

答案 0 :(得分:2)

  

通常,您不必明确指定CPU或GPU。 TensorFlow使用您的第一个GPU(如果有的话)进行尽可能多的操作。

G_1的输出将进入G-2,这也将在GPU上执行。

资源:Tensorflow Basic Usage