Hi, I currently have cuDNN7 installed on my machine. I presumed that I installed PyTorch based on that. When I try to run a piece of code, however, I get the following error:
File "/home/sam/.virtualenvs/cv/lib/python2.7/site-packages/torch/nn/functional.py", line 1011, in affine_grid
    return AffineGridGenerator.apply(theta, size)
  File "/home/sam/.virtualenvs/cv/lib/python2.7/site-packages/torch/nn/_functions/vision.py", line 75, in forward
    AffineGridGenerator._enforce_cudnn(theta)
  File "/home/sam/.virtualenvs/cv/lib/python2.7/site-packages/torch/nn/_functions/vision.py", line 66, in _enforce_cudnn
    assert cudnn.is_acceptable(input)
  File "/home/sam/.virtualenvs/cv/lib/python2.7/site-packages/torch/backends/cudnn/__init__.py", line 49, in is_acceptable
    if _libcudnn() is None:
  File "/home/sam/.virtualenvs/cv/lib/python2.7/site-packages/torch/backends/cudnn/__init__.py", line 25, in _libcudnn
    'but linked against {}'.format(compile_version, __cudnn_version))
RuntimeError: cuDNN version mismatch: PyTorch was compiled against 6021 but linked against 7001

It does sound like it needs cuDNN6, but isn’t there any option that I could use PyTorch together with cuDNN7?