I am currently running some tests with
lightning.gpu on a server with 4 gpus. I am wondering if there is a way to select which gpu to run on… perhaps something aking to
torch.device("cuda:x") where x corresponds to the number of the gpu you want to run on.
From my tests, I can see that the default behaviour is to run on gpu0, even though that might be at 100% usage or the vram is full.
Thanks in advance,