search

open-mmlab / mmaction2

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark
https://mmaction2.readthedocs.io
Apache License 2.0
4.06k stars 1.2k forks source link

When a computer has multiple GPUs, how to select GPUs for training? #2751

Open 15762260991 opened 8 months ago

15762260991 commented 8 months ago

The doc issue

When a computer has multiple GPUs, how to select GPUs for training?

Suggest a potential alternative/fix

No response

irvingzhang0512 commented 7 months ago

Training and Testing

CUDA_VISIBLE_DEVICES=0,1,2,3 PORT=29500 bash tools/dist_train.sh ${CONFIG} 4 [PY_ARGS]
CUDA_VISIBLE_DEVICES=4,5,6,7 PORT=29501 bash tools/dist_train.sh ${CONFIG} 4 [PY_ARGS]