Import of pytorch does not work

Robotics & Edge Computing Jetson Systems Jetson AGX Xavier
1

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 :)

3

Hi,

Thanks for reporting this issue to us.

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

4

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.

6

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

7

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