I am running the PyTorch python code and getting the error from windows 10 that shows "python has stopped working". when I run the same code on Ubuntu it returns t