Multi GPU Support?

[wchodev]

Hello, I've tried to train FlowGMM for the CIFAR10 dataset with the provided example code:
python3 experiments/train_flows/train_semisup_cons.py --[arguments]

When I try to train with multiple GPUs with the option --gpu_ids (e.g., --gpu_ids=0,1), the following error comes out: RuntimeError: expected device cuda:1 but got device cuda:0.

Any suggestion for multiple GPU training? or Is the code only support single GPU training?