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

post by sabalaba on May 15, 2018

sabalaba

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

4.0k views

post by sabalaba on May 15, 2018

sabalaba

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