How to specify which GPU to use for model training

[sanvritti]

How can I assign a specific GPU for training when I want to use only one GPU? I've already set 'use_gpu = True,' and I'm aware that when training on multiple GPUs, you can use the command python3 -m paddle.distributed.launch --gpus '0,1,2,3' tools/train.py Is there an equivalent option or flag to specify which GPU to use when conducting single-GPU training?