RuntimeError: cuDNN version mismatch: PyTorch was compiled against 7005 but linked against 7102

Technical Help
1

Lambda Stack Cuda pytorch error when getting the cudnn version with torch.backends.cudnn.versino(). OS: Ubuntu 16.04

user@hostname$ python3
Python 3.5.2 (default, Nov 23 2017, 16:37:01)
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.version
'0.3.1'
>>> torch.backends.cudnn.version()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/lib/python3/dist-packages/torch/backends/cudnn/init.py", line 34, in version
    if _libcudnn() is None:
  File "/usr/lib/python3/dist-packages/torch/backends/cudnn/init.py", line 27, in _libcudnn
    'but linked against {}'.format(compile_version, __cudnn_version))
RuntimeError: cuDNN version mismatch: PyTorch was compiled against 7005 but linked against 7102
2

Quick fix (until we can find out what issue is with Lambda Stack):

apt install python3-pip
pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision