우당탕
[Pytorch] 'DataLoader worker (pid(s) 00000) exited unexpectedly' or 'shared memory error' 오류 해결
캥거루
2021. 5. 13. 00:16
pytorch 환경에서 작업할때 종종 workers 를 gpu에 할당시킬 때 해당 오류가 발생한다.
Dataloader 와 연동되는 부분이라, DataLoader 세팅에서 파라미터를 num_workers=n 으로 할당했다면 num_workers=0 으로 바꿔주면 된다.
나 같은 경우엔 mmdetection을 쓰다가 발생한 오류라, cfg.data.workers_per_gpu = 0 으로 세팅해주니 해결됨.
관련 이슈 확인은 아래에서.