set multiple gpu using set_device()

[shwetagarg-dev]

I am unable to use multiple gpu using set_device. I have tried following things, but nothing working:

set_device(0,1)
set_device([0,1])

Please guide for proper usage.

[zjharper]

Python allows you to choose a single GPU using set_device(). Multi-GPU is only supported on the C++ interface. The --gpu flag used for this purpose is discussed in the caffe-docs.

[jiangqb]

Terrible...

[zjharper]

Update: Merge #4563 in January integrated NCCL to enable multi-GPU through the python interface. I suggest trying out the train.py demonstration. Good luck!

[mrgransky]

My trial as expined here did not work for TEST phase!

So how does the API work in python now for TEST phase?

Stop arrte de m'envoyer des message Le dim. 26 avr. 2020 11 h 20 a.m., Farid <notifications@github.com> a écrit :

…

so how does the API work in python now for test phase? — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub <#4253 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AIU36356NFHZ5ATZO7GEXR3RORGLBANCNFSM4CFOU4KQ> .