如何在Tensorflow中使用其他服务器上的GPU？

Question

我有两台服务器，每台服务器上都有多个GPU。它们在同一个局域网中。而且我可以使用SSH通过一个帐户连接这两者。

一个模型正在服务器1上训练，并且模型已保存在服务器1中。我想使用服务器2进行评估。服务器1上的模型有什么方法可以使用服务器2上的GPU？

据我所知，tf.device()可以分配GPU进行训练。是否可以分配192.168.1.2:/gpu:0之类的GPU？

谢谢！