Skip to content
This repository has been archived by the owner on Apr 17, 2023. It is now read-only.

Multi-GPU and CPU support #154

Draft
wants to merge 14 commits into
base: ote
Choose a base branch
from
Draft

Conversation

druzhkov-paul
Copy link

@druzhkov-paul druzhkov-paul commented Jul 30, 2021

To run sample several envvars should be defined:

MASTER_PORT - required; has to be a free port on machine with rank 0
MASTER_ADDR - required (except for rank 0); address of rank 0 node
WORLD_SIZE - required; can be set either here, or in a call to init function
RANK - required; can be set either here, or in a call to init function

Single GPU example:

$ MASTER_ADDR=localhost MASTER_PORT=2000 WORLD_SIZE=1 RANK=0 CUDA_VISIBLE_DEVICES=0 python mmdet/apis/ote/sample/sample.py configs/ote/custom-object-detection/mobilenetV2_ATSS/template.yaml

Two GPUs example:

$ MASTER_ADDR=localhost MASTER_PORT=2000 WORLD_SIZE=2 RANK=1 CUDA_VISIBLE_DEVICES=0,1 python mmdet/apis/ote/sample/sample.py configs/ote/custom-object-detection/mobilenetV2_ATSS/template.yaml &
$ MASTER_ADDR=localhost MASTER_PORT=2000 WORLD_SIZE=2 RANK=0 CUDA_VISIBLE_DEVICES=0,1 python mmdet/apis/ote/sample/sample.py configs/ote/custom-object-detection/mobilenetV2_ATSS/template.yaml

@druzhkov-paul druzhkov-paul self-assigned this Nov 23, 2021
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant