Import torch gives libcudnn OSError

vision
1

I have torch 1.8.0 installed. When I try to import it in python 3.8, it gives me the following error

import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/n/home00/<usr_name>/miniconda3/envs/mae/lib/python3.8/site-packages/torch/__init__.py", line 217, in <module>
    _load_global_deps()
  File "/n/home00/<usr_name>/miniconda3/envs/mae/lib/python3.8/site-packages/torch/__init__.py", line 177, in _load_global_deps
    raise err
  File "/n/home00/<usr_name>/miniconda3/envs/mae/lib/python3.8/site-packages/torch/__init__.py", line 172, in _load_global_deps
    ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
  File "/n/home00/<usr_name>/miniconda3/envs/mae/lib/python3.8/ctypes/__init__.py", line 373, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: libcudnn.so.8: cannot open shared object file: No such file or directory
2

Could you update PyTorch to the latest stable or nightly binary, as 1.8.0 is quite old by now?