RuntimeError: cuDNN version incompatibility: PyTorch was compiled against 7401 but linked against 7301

Since I was having issues with PyTorch 0.4.0, mainly with importing interpolate, I decided to install PyTorch 1.0.0. However when I run the train_network.py file, I get this error.

How do I fix this?

