Tensorflow estimator 多 GPU 预测

Jay*_*Yip 5 multi-gpu tensorflow tensorflow-estimator

我有一个估算器,可以通过指定train_distributeeval_distribute在配置中使用多个 GPU 进行训练和评估。但predict_distributetf.estimator.RunConfig. 如何使用多个 GPU 进行预测?