Import of pytorch does not work

Hey I am using the new Jetpack version 5 on the AGX and I tried to install pytorch v1.12.0. During the import of pytorch in python I receive the following error:
OSError: libcurand.so.10: cannot open shared object file: No such file or directory

I used the following commands to install pytorch:

wget https://developer.download.nvidia.com/compute/redist/jp/v50/pytorch/torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl -O torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install Cython
pip3 install numpy torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl

I am actually quite desperate at this point, because I am spending now hours to fix that. Since we need it quite fast, a fast reply would be awesome.

Kindly regards and have a great day :)

Hi,

Thanks for reporting this issue to us.

We are testing this internally.
Will share more information with you later.

Hi,

Sorry that we cannot reproduce this issue internally.
The PyTorch works well in our environment.

$ python3
Python 3.8.10 (default, Mar 15 2022, 12:22:08)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> 

Could you share the output of the below command first?

$ sudo apt show nvidia-jetpack

Thanks.

Issue is fixed, I reinstalled JetPack. I assume I forgot to set the tick for the Cuda installation. Stupid mistake, I am so sorry.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.