How to install torchvision on jetson pack 5.1.5 orin

How to install torchvision on Jetson. I am using Jetson 5.1.5.

Python 3.8.20 (default, Oct 3 2024, 15:18:56)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type “help”, “copyright”, “credits” or “license” for more information.

import torch
import torchvision
print(f"PyTorch version: {torch.version}“)
PyTorch version: 2.1.0a0+41361538.nv23.06
print(f"Torchvision version: {torchvision.version}”)
Torchvision version: 0.16.0+fbb4cc5

Above are versions of Torch and TorchVision. I installed torchvision via source build. but i got errors

warnings.warn(
/home/wy350/miniconda3/envs/rebar-v2/lib/python3.8/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: '/home/wy350/miniconda3/envs/rebar-v2/lib/python3.8/site-packages/torchvision/image.so: undefined symbol: _ZN5torch3jit17parseSchemaOrNameERKSsb’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?

How to install corresponding Torchvision with Torch?

Hi,

We have prebuilt package in the below link:

Please give it a try.
Thanks.

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