VinAIResearch/3D-UCaps

How to enable all gpus?

Closed this issue · 2 comments

Even when i choose --gpus = 0,1 and --gpus=2 , it is considering only one GPU for training. I have two gpus and wanted to use them both. Please let me know how to enable my both GPUs

Can you show me the log of your run? Pytorch lightning will print out which devices you are using in that run. You are using the right argument so I do not know what causes your problem.

Thank you ..its working as you said. There were some issues at my end