set multiple gpu using set_device()

[shwetgarg]

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?

[ghost]

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> .