Tensorflow 多GPU预测的张量流估计器

我有一个估计器，可以通过在配置中指定

train\u distribute

和

eval\u distribute

来使用多个GPU进行训练和评估。但是在

tf.estimator.RunConfig

中没有

predict\u distribute

参数。如何使用多个GPU进行预测

你试过tf.contrib.estimator.replicate_model_fn吗？没有，因为

tf.contrib.estimator.replicate_model_fn

已被弃用。@JayYip你找到解决方案了吗？