It seems to be somehow related to your OS.
Could you pose some information of your system?
NVIDIA driver, CUDA version, CuDNN?, PyTorch version
This could help somehow.
What kind of error did you get after running Pytorch again?