Welcome to Jetson AGX Thor repository

03.Torchvision

Setting up the Torchvision Environment

Setting up the Torchvision Environment1. System Information2. Installing Dependencies3. Installing Torchvision3.1. Offline Installation3.2. Online Installation4. Verifying the Installation

1. System Information

2. Installing Dependencies

torchvision and torch require corresponding versions, so please install torch first according to the "Setting up the PyTorch Environment" tutorial, and then install torchvision.

Version Compatibility Lookup:


https://github.com/pytorch/vision#installation

`torch`	`torchvision`	Python
`main` / `nightly`	`main` / `nightly`	`>=3.10`, `<=3.14`
`2.10`	`0.25`	`>=3.10`, `<=3.14`
`2.9`	`0.24`	`>=3.10`, `<=3.14`
`2.8`	`0.23`	`>=3.9`, `<=3.13`
`2.7`	`0.22`	`>=3.9`, `<=3.13`
`2.6`	`0.21`	`>=3.9`, `<=3.12`

Based on the version compatibility matrix between PyTorch and Torchvision, since the installed version of Torch is 2.10, you need to install Torchvision version v0.25.

3. Installing Torchvision

You may choose either the offline or the online method to install Torchvision.

3.1. Offline Installation

Manually visit the download URL to download the latest version of Torchvision.

Download URL: https://pypi.jetson-ai-lab.io/sbsa/cu130/+f/62e/2fd4aaade43f3/torchvision-0.25.0-cp312-cp312-linux_aarch64.whl#sha256=62e2fd4aaade43f398d7ab31515b4a05c4cc8428d54ab248caa526c155ffc88e


xxxxxxxxxx
cd Downloads/


xxxxxxxxxx
sudo pip3 install torchvision-0.25.0-cp312-cp312-linux_aarch64.whl --break-system-packages

3.2. Online Installation


xxxxxxxxxx
sudo pip3 install torchvision --break-system-packages --index-url https://pypi.jetson-ai-lab.io/sbsa/cu130

4. Verifying the Installation


xxxxxxxxxx
python3 -c "import torchvision; print(f'Torchvision: {torchvision.__version__}')"

PyTorch

Torchaudio