JetPack 6.1 & CUDA 12.6 Support cuSPARSELt (Pytorch installation)

Hi there!

Brief Overview of the Issue

I am trying to install Pytorch along with ‘torchvision’ with cuda support on my Jetson Orin Nano Developer Kit, however when I download ‘torch’ from this torch download .whl for jp61 link and go on to verify the installation using ‘python3 -c “import torch”’ I encounter the following error:

Encountered ERROR

Traceback (most recent call last): File "<string>", line 1, in <module> File "/usr/local/lib/python3.10/dist-packages/torch/__init__.py", line 361, in <module> from torch._C import * # noqa: F403 ImportError: libcusparseLt.so.0: cannot open shared object file: No such file or directory

torchvision package

Moreover it seems that ‘torchvision’ isn’t even available for download for JetPack 6.1 in the link,
How can I install a ‘torchvision’ version which would be compatible with ‘torch’ with cuda support?

Environment Info

Environment:

  • JetPack 6.1
  • CUDA 12.6
  • Python 3.10

cuSPARSELt Support CUDA 12.6

I wondered if cuSPARSELt is even supported by CUDA 12.6 yet, is it?
Since I didn’t find any helpful information on the matter. . .
Can anyone shine some light on this subject or even suggest some solutions please?

Thanks in advance for anyone stopping by to help! :D

1 Like

Install cuSPARSELt from https://developer.nvidia.com/cusparselt-downloads and torchvision 0.20 from https://github.com/pytorch/vision