mmaction2 icon indicating copy to clipboard operation
mmaction2 copied to clipboard

Published 20 hours ago • verified publisher icon open-mmlab

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

Open 15762260991 opened this issue 1 year ago • 1 comments

The doc issue

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

Suggest a potential alternative/fix

No response

15762260991 avatar Nov 17 '23 02:11 15762260991

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]

irvingzhang0512 avatar Nov 27 '23 08:11 irvingzhang0512