What is proper version of torchvision for torch 2.0.0+nv23.5

[dusty_nv]

@yys9905a

i use script ‘pip install torchvision==0.14.1’ for install torchvision something wrong my script?

@yys9905a that way will install torchvision without CUDA support, please uninstall it first and then see the Installation section of this post:

PyTorch for Jetson

Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). You can also use the pre-built l4t-pytorch and l4t-ml container images and Dockerfiles. PyTorch pip wheels JetPack 5

PyTo…

Or as Aasta suggested, you can run l4t-pytorch container if you are still having problems.