search

davidsandberg / facenet

Face recognition using Tensorflow
MIT License
13.71k stars 4.8k forks source link

How can I train with multi GPU? #797

Open morndew410 opened 6 years ago

morndew410 commented 6 years ago

I executed "train_softmax.py" with 8 GPU machine but it only used GPU 0. How can I execute with multi-GPU?

Thanks.

image

look4pritam commented 6 years ago

You will have to modify code and provide functionality like TensorFlow-Slim for multi GPU support.

soldier828 commented 6 years ago

ops