How to specify the code to run on a particular gpu device

[ikun0124]

I want to run a code on gpu 0 and another code on gpu 1, how should I set up the config file？

[ikun0124]

need help.......

[ikun0124]

need help.......

[samet-akcay]

Hi @loyal-ikun, you could refer to this link regarding configuring the GPUs.
https://lightning.ai/docs/pytorch/stable/accelerators/gpu_basic.html#train-on-gpus

Based on this, you will need to modify accelerator and devices arguments in Trainer configurations in a config file.

If you have any more questions that you cannot find in the documentation, let us know.