[PYTHON] RuntimeError: DataLoader worker (pid (s) ~) Cause and solution for exited unexpectedly

Execution environment

When I try to use DataLoader with Pytorch, I get the following error

Program executed


torch.utils.data.DataLoader(
                            dataset = dataset,
                            batch_size = 100,
                            shuffle = True,
                            num_workers = 8)

Error statement


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

Cause

Due to lack of memory

solution

Reduce num_workers to a value that does not cause an error ex) num_workers = 0 etc.

Recommended Posts

RuntimeError: DataLoader worker (pid (s) ~) Cause and solution for exited unexpectedly