How to disable using two GPUs for training?

[davidvct]

I have two GPUs on the computer. I am training SwinIR for denoising, it ran on two GPUs, even though I specified one or no GPU in the json file. I also tried setting the 'dist' to false, but it still ran on both GPUs.

I tried to specified which GPU to use in main_train_psnr.py as well, but got the same result:

def main():
    print("----project SwinIR------")

    os.environ["CUDA_DEVICES_ORDER"] = "PCI_BUS_ID"
    os.environ["CUDA_VISIBLE_DEVICES"] = '1'

I want to use only 1 GPU, if I can specified which GPU that would be nice.

Thanks.