Running On Multiple GPUs about consingan

[tohinz]

Without seeing the code it's difficult.
Have you changed how the parameter --gpu is handled (in the main_train.py file)?
By default it's set to 0 and later in the code we do (see here)
if torch.cuda.is_available(): torch.cuda.set_device(opt.gpu)
You might have to change that to get it to work.

[cbarne10]

Hi, still having the problem. I changed that line to equal a torch device of two gpus (passed in set_device). It still runs on 1 gpu.

[tohinz]

Sorry for the late response.
What kind of GPU are you running this on and how much VRAM does it have?
I run all of my experiments on a single GPU with ~12GB VRAM without problems.

[cbarne10]

GPU 0 with about 30000 MiB

[Liz1317]

I have the same problem. Is there any solution?