인프런 커뮤니티 질문&답변
작성한 질문수
제 연구실 gpu 여러개로 모델을 학습시키고 싶습니다(mask-rcnn)
작성
·
303
0
안녕하세요 강의 감사합니다.
저희 랩실 gpu를 이용하여 모델을 돌려보고 싶은데
os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID' os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
이런 식으로 진행하면 될까요? 아니면 config에 수정할 부분이 있을까요?
일단 코랩으로만 진행해 보았는데 한 번 수정해보겠습니다. 감사합니다./
답변 1
1
권 철민
지식공유자
안녕하십니까,
저도 multi gpu로 mmdetection을 적용해 보지는 않았지만, 과거 버전에서 multi gpu가 이슈가 있어서 현재는 더 어려운 방식으로 좀 제공해 주는것 같습니다.
아래의 dist_train.sh을 이용해야만 하는데,
https://github.com/open-mmlab/mmdetection/blob/master/tools/dist_train.sh
https://mmdetection.readthedocs.io/en/v2.2.1/getting_started.html 에서 Train with multiple GPUS 부분을 참조하셔서 dist_train.sh을 구동 시키셔야 합니다.
감사합니다. 현재 주피터에 연결하는 문제를 해결한다고 시간이 좀 걸렸는데 관련해서 작업 진행해 보겠습니다. 감사합니다