7-2 TorchVision Environment Setup

Initial environment setup

sudo apt update && \
sudo apt install nvidia-jetpack -y && \
sudo apt install python3-pip -y

Install and test TorchVision

Verify available versions for installation

// Verify available versions
pip3 index versions torchvision -i https://pypi.jetson-ai-lab.io/jp6/cu126

// Install a specific version (this tutorial uses v0.23.0)
pip3 install torchvision==0.23.0 -i https://pypi.jetson-ai-lab.io/jp6/cu126

// Test the installation
python3 -c 'import torchvision; print(torchvision.__version__);'

 

Copyright © 2026 YUAN High-Tech Development Co., Ltd.
All rights reserved.