Installing Pytorch OSError: libcurand.so.10: cannot open shared object file: No such file or directory

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Xavier
9

I have the same problem on jetson xavier (jetpack 4.4-b144)

After installing torch-1.6.0-cp36-cp36m-linux_aarch64.whl following this manual PyTorch for Jetson

import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/discort/test/.env/lib/python3.6/site-packages/torch/__init__.py", line 188, in <module>
    _load_global_deps()
  File "/home/discort/test/.env/lib/python3.6/site-packages/torch/__init__.py", line 141, in _load_global_deps
    ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
  File "/usr/lib/python3.6/ctypes/__init__.py", line 348, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: libcurand.so.10: cannot open shared object file: No such file or directory

Cuda version: 10.2

nvcc --version
Cuda compilation tools, release 10.2, V10.2.89

How did you manage to solve this issue?

@dusty_nv @ho126jin