Hey @CCL, you will need to set the CUDA_AVAILABLE_DEVICES env var before launching the process. Sth like:
$ CUDA_AVAILABLE_DEVICES=0 python main.py

If you just want to set the default device, you can use set_device 209
Update
Please ignore the code above. I miss-read the variable name. It should be CUDA_VISIBLE_DEVICES