How to specify GPU for single machine multi-gpus training

[zhaijianyang]

How to specify GPU for single machine multi-gpus training?

[irvingzhang0512]

You could use --gpu-ids

mmaction2/tools/train.py

Lines 37 to 42 in ee34d95

	group_gpus.add_argument(
	'--gpu-ids',
	type=int,
	nargs='+',
	help='ids of gpus to use '
	'(only applicable to non-distributed training)')

But I prefer setting CUDA_VISIBLE_DEVICES

CUDA_VISIBLE_DEVICES=1 python tools/train.py xxx

[congee524]

Hi, you can also refer to launch-multiple-jobs-on-a-single-machine

[zhaijianyang]

You could use --gpu-ids

mmaction2/tools/train.py

Lines 37 to 42 in ee34d95

	group_gpus.add_argument(
	'--gpu-ids',
	type=int,
	nargs='+',
	help='ids of gpus to use '
	'(only applicable to non-distributed training)')

But I prefer setting CUDA_VISIBLE_DEVICES

CUDA_VISIBLE_DEVICES=1 python tools/train.py xxx

ok,thank you