How may I select which cuda device I use?

[Siigari]

I have two GPUs. How do I command line select which one to use?

Thanks.

[jiwoo-writing]

I inserted this on top of core.py

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "1"