Cannot install PyTorch on Orin AGX, with JP 5.0.2

[gbetsos]

I have a custom Orin AGX board with L4T 35.1.0 and JetPack 5.0.2GA.

I followed all steps from page: Installing PyTorch for Jetson Platform

export TORCH_INSTALL=https://developer.download.nvidia.com/compute/redist/jp/v502/pytorch/torch-1.13.0a0+d0d6b1f2.nv22.10-cp38-cp38-linux_aarch64.whl

$ python3 -m pip install --upgrade pip; 

$ python3 -m pip install aiohttp numpy=='1.19.4' scipy=='1.5.3' 

$ export "LD_LIBRARY_PATH=/usr/lib/llvm-8/lib:$LD_LIBRARY_PATH"; 

$ python3 -m pip install --upgrade protobuf; 

$ python3 -m pip install --no-cache $TORCH_INSTALL

After trying to do import torch I get the following error message:

Traceback (most recent call last):
File “”, line 1, in
File “/home/nvidia/envs/torch/lib/python3.8/site-packages/torch/init.py”, line 191, in
_load_global_deps()
File “/home/nvidia/envs/torch/lib/python3.8/site-packages/torch/init.py”, line 153, in _load_global_deps
ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
File “/usr/lib/python3.8/ctypes/init.py”, line 373, in init
self._handle = _dlopen(self._name, mode)
OSError: libnvToolsExt.so.1: cannot open shared object file: No such file or directory

[dusty_nv]

Hi @gbetsos, do you have CUDA Toolkit installed on your Jetson? My guess is perhaps not, since you said it’s a custom system.

find /usr/local -name 'libnvToolsExt.so*'

/usr/local/cuda-11.4/targets/aarch64-linux/lib/libnvToolsExt.so.1.0.0
/usr/local/cuda-11.4/targets/aarch64-linux/lib/libnvToolsExt.so
/usr/local/cuda-11.4/targets/aarch64-linux/lib/libnvToolsExt.so.1

If you don’t have it installed, try installing it with sudo apt-get update && sudo apt-get install cuda-toolkit-11-4