通过ssh在本地计算机上运行的Tensorflow和Caffe

Question

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

Answer 1

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