Skip to main content

7-2 TorchVision Environment Setup

PANDORA DEVELOPER GUIDE · CHAPTER 07

Initial environment setup ➤ For the installation of JetPack and Python-related tools, you can review: [ Chapter 2-3: Installing the Jetson Component Environment ]

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__);'

Adapted from the original YUAN Developer Hub article. Source ID: 256.