how to assign specific gpus in mmdetection during training?

[GolvenDong]

The documentation says I can use dist_trian.sh to train with mutiple gpus,but how to specify gpus? For example,I want to use gpu:2,3 to train my models.
The official tutorial can only specify the gpu numbers.

[jbwang1997]

You can add CUDA_VISIBLE_DEVICES="2,3" in the front of dist_trainsh

[ZwwWayne]

Note that we can add CUDA_VISIBLE_DEVICES=2.3 in both dist_train.sh and slurm_train.sh.

[GolvenDong]

Okay，I will try it!
thx!