인프런 커뮤니티 질문&답변

munhaneul2208님의 프로필 이미지
munhaneul2208

작성한 질문수

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

tiny kitti 데이터로 MMDetection Train 실습 - Train 실행 후 이미지 Inference

맥북 환경

작성

·

211

0

안녕하세요 강의 잘 듣고 있습니다!

제가 지금 맥북 로컬 환경에서 코드를 실행 시켜려고 하는데 cfg.device = 'cpu' 로 설정하고 

train_detector(model, datasets, cfg, distributed=False, validate=True)

를 실행했을 때 아래와 같은 오류가 발생합니다.

RuntimeError: DataLoader worker (pid(s) 11663) exited unexpectedly 

로컬 환경에서 돌리기 위해서는 nvidia GPU 컴퓨터에서 cuda를 따로 설치하여야 이번 강의 코드를 실행 시킬 수  있는 건가요?  cuda 관련하여 아직 지식이 많이 부족합니다 ㅜ 

혹시 맥북에서 cpu를 사용하여 실행 시킬 수 있는 방법이 없을 까요? 

답변 1

0

권 철민님의 프로필 이미지
권 철민
지식공유자

안녕하십니까, 

맥북 CPU에서 잘 안되는 군요. 

그런데 문제없이 잘 되어도, CPU로 학습을 시켜려면 굉장히 오랜 시간이 걸립니다.  GPU로 1시간 걸리면 CPU로 1~2일 이상 걸릴 수 있습니다. 

현실적으로 CPU로 학습을 시키는 것은 무리입니다. 

Inference는 CPU로 시간이 걸릴 수도 있지만 어느정도 가능할 수 있습니다. 

아래는 CPU로 Inference를 수행한 자료입니다. 

https://bluehorn07.github.io/2021/01/26/mmdetection-cpu.html

꼭 CPU로 학습을 하셔야 겠다면, 제가 크게 도와드리지 못할 수도 있지만, 어쩧든 위의 에러 메시지는 데이터 로딩 과정에서 오류가 발생하니 일단 학습 데이터 세트가 정확히 잘 설정되었는지 먼저 확인 부탁드립니다. 

감사합니다. 

munhaneul2208님의 프로필 이미지
munhaneul2208

작성한 질문수

질문하기