Using Jetson AGX Orin with Provided Ubuntu 20.04 System
We broadly follow these instructions:
Power Up
Connect USB keyboard and mouse, DisplayPort port, and USB Type-C power cable

Install Jetpack Tools
First, use this command to check board support package (BSP) version
cat /etc/nv_tegra_release
As seen in the output, the BSP is up-to-date, the apt sources lists are the latest, and we can proceed

Run the commands to install Jetpack
sudo apt update
sudo apt dist-upgrade
sudo reboot
sudo apt install nvidia-jetpack
Add the following lines to ~/.bashrc
...
# CUDA
export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
...
Now it should detect CUDA
nvcc --version

Install PyTorch
Install conda according to the "Install Conda" section in this guide
Quick Start With Chipyard on Ubuntu or WSLconda create -yn tensorrt python=3.10
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
Last updated
Was this helpful?