Using Jetson AGX Orin with Provided Ubuntu 20.04 System

We broadly follow these instructions:

Getting Started with Jetson AGX Orin Developer KitNVIDIA Developer

Jetson AGX Orin Developer Kit User Guide - Hardware LayoutNVIDIA Developer

Quick Start — NVIDIA Jetson Linux Developer Guide 1 documentation

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

.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 WSL

conda create -yn tensorrt python=3.10

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124