I only have one GPU

[sabhiram6]

Hey @tobran I have read your

paper and its excellent . I am currently implementing the code . I only have one GPU so in train.sh i changed the nodes to 1 and set CUDA_VISIBLE_DEVICES=0 .

But i am getting " train.py: error: unrecognized arguments: --local-rank=0" .

Could you please tell me what changes i need to make in the code if I have access to only 1 GPU .

Thank You

[tobran]

Thank you for your interest in our work. You can check the pytorch version, if it is more than 2.0, you will encounter this error

[sabhiram6]

Okay thank you. I used torchrun and now it is working .

[VinayHajare]

Okay thank you. I used torchrun and now it is working .

Can you please help me with training on single GPU, I have tried many times also with torchrun. Please tell what changes did you do with the local_rank

[sabhiram6]

hey Vinay , I made some small changes like setting MultiGpu to False everywhere and only gave one GPU id in yml files. And later adjusted those according to torchrun .

…

On Sun, Feb 11, 2024 at 10:22 AM Vinay Arjun Hajare < ***@***.***> wrote: Okay thank you. I used torchrun and now it is working . Can you please help me with training on single GPU, I have tried many times also with torchrun. Please tell what changes did you do with the local_rank — Reply to this email directly, view it on GitHub <#21 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AWBIGNEPSBAC2L4FAGB3PW3YTBE7LAVCNFSM6AAAAABBGJHKY2VHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMYTSMZXGQZDIOJWGM> . You are receiving this because you modified the open/close state.Message ID: ***@***.***>