get_jetson_files.sh

#!/bin/bash
# Get the Jetson BSP and rootfs, then prepare for flashing

# 
# MIT License
# Copyright (c) 2021, JetsonHacks
#

# Sanity warning; Make sure we're not running from a Jetson
# First check to see if we're running on Ubuntu
# Next, check the architecture to make sure it's not aarch64, not a Jetson
JETSON_FOLDER=R32.6.1
XAVIER_16=false


function help_func
{
	echo "Usage: ./get_jetson_files.sh [OPTIONS]"
	echo "   No option Downloads 32.6.1 L4T VERSION for xavier files"
	echo "   --xavier-16 Adds Overlay to support Jetson Xavier NX 16GB"
	echo "   -h | --help - displays this message"
}

if [ -f /etc/os-release ]; then
  if [[ ! $( grep Ubuntu < /etc/os-release ) ]] ; then
    echo 'WARNING: This does not appear to be an Ubuntu machine. The script is targeted for Ubuntu, and may not work with other distributions.'
    read -p 'Continue with installation (Y/n)?' answer
    case ${answer:0:1} in
       y|Y )
         echo Yes
       ;;
       * )
         exit
       ;;
    esac

  else
    if [ $(arch) == 'aarch64' ]; then
      echo 'This script must be run from a x86 host machine'
      if [ -f /etc/nv_tegra_release ]; then
	 echo 'A aarch64 Jetson cannot be the host machine'
      fi
      exit
    fi
  fi
else
    echo 'WARNING: This does not appear to be an Ubuntu machine. The script is targeted for Ubuntu, and may not work with other distributions.'
    read -p 'Continue with installation (Y/n)?' answer
    case ${answer:0:1} in
       y|Y )
         echo Yes
       ;;
       * )
         exit
       ;;
    esac
fi

while [ "$1" != "" ];
do
   case $1 in
   --xavier-16 )
		XAVIER_16=true
		;;
	-h | --help )
		help_func
		exit
	  ;;
	* )
		echo "*** ERROR Invalid flag"
		help_func
		exit
	   ;;
	esac
	shift
done


echo 'Ready to download!'
mkdir $JETSON_FOLDER
cd $JETSON_FOLDER

# Made it this far, we're ready to start the downloads

# Get the R32.6.1 Tegra system
# Get the L4T Driver Package - BSP
wget -N https://developer.nvidia.com/embedded/l4t/r32_release_v6.1/t186/jetson_linux_r32.6.1_aarch64.tbz2
# Get the Sample Root File System (rootfs)
wget -N https://developer.nvidia.com/embedded/l4t/r32_release_v6.1/t186/tegra_linux_sample-root-filesystem_r32.6.1_aarch64.tbz2
# Get the Secure Boot package
wget -N https://developer.nvidia.com/embedded/l4t/r32_release_v6.1/t186/secureboot_r32.6.1_aarch64.tbz2

# Unpack the files, creating the Linux_for_Tegra folder
sudo tar xpvf jetson_linux_r32.6.1_aarch64.tbz2

if [[ "$XAVIER_16" = true ]]; then
  # get Overlay files  to support Jetson Xavier NX 16GB
  wget -N  https://developer.nvidia.com/xnx-16gb-r3261-overlaytbz2
  tar xpvf xnx-16gb-r3261-overlaytbz2
  tar xf xnx-16gb-r32.6.1-overlay.tbz2
fi

cd Linux_for_Tegra/rootfs/
sudo tar xpvf ../../tegra_linux_sample-root-filesystem_r32.6.1_aarch64.tbz2
cd ../..
tar xvjf secureboot_r32.6.1_aarch64.tbz2
cd Linux_for_Tegra/
# The NVIDIA scripts do not officially support Ubuntu 20.04 on the host
# Set the LDK_ROOTFS_DIR enviornment variable to compensate
if [[ $(lsb_release -rs) == "20.04" ]] ; then
  export LDK_ROOTFS_DIR=$PWD
fi
sudo ./apply_binaries.sh