Solving Torch Errors

Undefined `__nvJitLinkComplete_12_4`

Error

>>> import torch
  File "/home/tk/Documents/mambaforge/envs/torch/lib/python3.10/site-packages/torch/__init__.py", line 367, in <module>
    from torch._C import *  # noqa: F403
ImportError: /home/tk/Documents/mambaforge/envs/depth-pro/lib/python3.10/site-packages/torch/lib/../../nvidia/cusparse/lib/libcusparse.so.12: undefined symbol: __nvJitLinkComplete_12_4, version libnvJitLink.so.12

Solution

Add the nvjtlink directory of the corresponding python package to link library path.

export LD_LIBRARY_PATH=$CONDA_PREFIX/lib/python3.10/site-packages/nvidia/nvjitlink/lib/:$LD_LIBRARY_PATH