Orin AGX pyTorch SDK 513

Autonomous Machines Jetson & Embedded Systems Jetson AGX Orin
4

I have a similar issue with Jetpack 5.1.2 and torch 2.1.0a0+41361538.nv23.06.

I cloned torchvision and checkout v0.16.0.

FORCE_CUDA=1 sudo python3 setup.py install

I get error during import:

import torchvision
/usr/local/lib/python3.8/dist-packages/torchvision-0.16.0+fbb4cc5-py3.8-linux-aarch64.egg/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: '/usr/local/lib/python3.8/dist-packages/torchvision-0.16.0+fbb4cc5-py3.8-linux-aarch64.egg/torchvision/image.so: undefined symbol: _ZNK3c1017SymbolicShapeMeta18init_is_contiguousEv’If you don’t plan on using image functionality from torchvision.io, you can ignore this warning. Otherwise, there might be something wrong with your environment. Did you have libjpeg or libpng installed before building torchvision from source?
warn(
torchvision.version
‘0.16.0+fbb4cc5’
import torch
torch.version
‘2.1.0a0+41361538.nv23.06’