search

open-mmlab / mmaction2

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

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

Open 15762260991 opened 1 year ago

15762260991 commented 1 year 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 1 year 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]