Tensorflow 和 Caffe 通过 ssh 在本地机器上运行

Question

我正在尝试实现一个系统，它使用两个 CNN，一个用于人体检测，另一个用于姿势估计。第一个为第二个提供边界框，然后对于每个检测到的人，我得到一个预测姿势作为输出。问题是第一个网络的实现是在 Caffe 中，第二个是在 Tensorflow 中，所以我不能使用我机器上的单个 gpu 来运行两个网络 (https://github.com/tensorflow/tensorflow/issues/2210)。所以我在想是否可以在我的本地网络中的 gpu 上运行两个网络之一（也许通过 ssh？），你知道这是否可能，或者是否有一个框架可以类似的东西？

Answer 1

您可以在 GPU 上运行第一个网络，然后您必须以某种方式将生成的数据传输到具有另一个 GPU 的另一台计算机（如果您想这样做，也可以只使用 CPU）。您可以让您的第一个网络将结果保存到一个文件中，然后编写一个脚本将该文件（可以使用 scp）传输到另一台计算机。