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