[Pytorch] RuntimeError: DataLoader worker (pid(s) 19106, 19107, 19109, 19110) exited unexpectedly.

반응형

- 전체 에러 문구

RuntimeError: DataLoader worker (pid(s) 19106, 19107, 19109, 19110) exited unexpectedly

​

확실하지 않지만, dataloader에서 데이터를 불러올 때, 이미 tensor 형태로 데이터를 만들어 불러 오는 경우 발생하는 에러인 것으로 판단됩니다.

좀 더 구글링을 해보니 이미 tensor형태로 만들어진 데이터는 gpu에 들어가 있는 상태인데,  이를 다시 불러와 다른 gpu에 넣으려 할 때 발생하는 에러인 것 같습니다.

​

​

​

- 해결 방법

데이터를 넘겨줄 때 tensor로 만들어 넘겨 주는 방식 대신, numpy 형식으로 바꿔 넘겨주는 방식으로 해결하였습니다.