How to run Tensorflow Estimator on multiple GPUs with data parallelism

后端 未结 5 854
青春惊慌失措
青春惊慌失措 2021-01-31 06:17

I have a standard tensorflow Estimator with some model and want to run it on multiple GPUs instead of just one. How can this be done using data parallelism?

I searched

5条回答
  •  旧时难觅i
    2021-01-31 06:55

    You can find an example using tf.distribute.MirroredStrategy and tf.estimator.train_and_evaluate here.

提交回复
热议问题