Hello everyone,
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,
Patrick.