Hard to tell without a proper error message. So far it seems your IPython kernel dies, so it might be related to PyTorch.