How to run on multiple GPUs?

Open marsggbo opened this issue 6 years ago • 2 comments

Could you share some example codes that can run on multiple GPUs?

May 28 '19 10:05 marsggbo

@marsggbo: For multiple GPUs on the same machine depends on https://github.com/tensorflow/adanet/issues/87 being fixed first. However, you can distribute training across multiple machines each with one GPU using TF_CONFIG: https://cloud.google.com/ml-engine/docs/tensorflow/distributed-training-details

You can also look at https://github.com/tensorflow/adanet/tree/master/research/improve_nas to see how we ran some experiments on Google Cloud with multiple GPUs.

Jun 04 '19 18:06 cweill

@marsggbo: For some examples of multiple GPUs clink on the below link I think that helps you :) https://jhui.github.io/2017/03/07/TensorFlow-GPU/

Sep 08 '19 02:09 gargprince1702