custom_image_classifier_pytorch icon indicating copy to clipboard operation
custom_image_classifier_pytorch copied to clipboard
verified publisher icon amrit-das

How can I train in multi GPU?

Open Laudarisd opened this issue 5 years ago • 0 comments

Hello, Thank you for the great frame work. I have question regarding GPU. Currently I am using a server which contains 8 GPUs. How can I use all of them while training?

I tried to use CUDA_VISIBLE_DEVICES=0,1,2,.... but didn't work. It only uses 1 gpu. Thank you.

Laudarisd avatar Mar 24 '20 02:03 Laudarisd