报错：RuntimeError: DataLoader worker (pid(s) 10188) exited unexpectedly

Pytorch 报错：RuntimeError: DataLoader worker (pid(s) 2420, 11236, 7252, 3104) exited unexpectedly

并没有显示出是哪一行代码出错，但是根据提示中的DataLoader worker可知是多线程的问题，对应到代码中应该就是下图这里：

num_workers = 4
    train_iter = torch.utils.data.DataLoader(mnist_train, batch_size=batch_size, shuffle=True, num_workers=num_workers)
    test_iter = torch.utils.data.DataLoader(mnist_test, batch_size=batch_size, shuffle=False, num_workers=num_workers)

将num_worker改成1就可以了。