우당탕

[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 으로 세팅해주니 해결됨. 

 

관련 이슈 확인은 아래에서.

 

github.com/pytorch/pytorch/issues/5301