inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

제 연구실 gpu 여러개로 모델을 학습시키고 싶습니다(mask-rcnn)

372

비밀

작성한 질문수 9

0

안녕하세요 강의 감사합니다.
저희 랩실 gpu를 이용하여 모델을 돌려보고 싶은데
 
os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID' os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
 
이런 식으로 진행하면 될까요? 아니면 config에 수정할 부분이 있을까요?
일단 코랩으로만 진행해 보았는데 한 번 수정해보겠습니다. 감사합니다./

딥러닝 python 머신러닝 배워볼래요? keras tensorflow 컴퓨터-비전

답변 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을 구동 시키셔야 합니다. 

0

비밀

감사합니다. 현재 주피터에 연결하는 문제를 해결한다고 시간이 좀 걸렸는데 관련해서 작업 진행해 보겠습니다. 감사합니다

강의 환경설정 질문

0

43

2

Custom Dataset에서의 polygon 정보 관련

0

86

3

cvat.ai 보안 수준이 궁금합니다

0

81

2

캐클 nucleus 챌린지 runpod 실습 코드 에러 질문드립니다.

0

98

3

추론 결과의 Precision(또는 mAP) 평가 방법

0

86

2

mmdetection mask rcnn inferenct 실습 시 runpod 템플릿 관해서 질문드립니다.

0

61

2

runpod에서 google drive 연결 시 오류 발생

0

110

2

로드맵 선택

0

68

1

mmcv

0

60

2

Anchor box의 Positive 처리 위치

0

63

2

해당 강의 runpod 적용 후 에러 제보드립니다

0

87

2

run pod credit 관련 제보

0

101

2

mmdetection 2.x과 3.x 호환 관련 표기

0

79

2

mm_faster_rcnn_train_kitti.ipynb 실행 오류

0

101

3

질문 드립니다.

0

83

3

mm_faster_rcnn_train_coco_bccd 실행 오류 질문드립니다.

0

81

1

강사님께 수정을 제안드리고 싶은 것이 있습니다.

0

96

1

google automl efficientdet 다운로드 및 설치 오류

0

76

1

이상 탐지에 사용할 비전 기술 조언 부탁드립니다.

0

104

2

OpenCV 관련 질문드립니다.

0

74

2

mmcv 설치관련해서 문의드려요

0

335

3

강의 구성 관련해서 질문이 있습니다

1

139

2

모델 변환 성능 질문드립니다.

0

123

1

NMS 로직 문의 드려요

0

116

2