基于数据并行实现多GPU跑模型预测(VGG16示例)

个人总结能实现多GPU跑图的方法: 1、使用谷歌框架tf.estimator; session_config = tf.ConfigProto(device_count={'GPU': 0,'GPU':1,'GPU':2,'GPU':3}) run_config = tf.estimator.RunConfig().replace(session_config=session_config) estimator = tf.estimator.Estimator( model_fn=model_fn, model_dir=FLAGS.model_dir, config=run_co