Detect ARM64 + CUDA (Jetson) in install.sh and bootstrap.sh

install.sh: prompts user to choose CUDA vs CPU when nvidia-smi detected on aarch64
bootstrap.sh: auto-selects aarch64-cuda asset when NVIDIA GPU detected
Both rely on bundled libtorch/ in the release zip (no separate download needed)

Author: Michael Yuan

install.sh

@@ -103,7 +103,30 @@ resolve_asset() {
                 ASSET_NAME="asr-linux-x86_64"
             fi
             ;;
-        linux-aarch64)  ASSET_NAME="asr-linux-aarch64"   ;;
+        linux-aarch64)
+            if [ -n "$CUDA_DRIVER" ]; then
+                echo ""
+                info "NVIDIA GPU detected on ARM64 (Jetson). Choose build variant:"
+                echo "  1) CUDA 12.6  (recommended for Jetson)"
+                echo "  2) CPU only"
+                echo ""
+
+                local choice
+                read -r -p "Select variant [1]: " choice </dev/tty
+                choice="${choice:-1}"
+
+                case "$choice" in
+                    1) USE_CUDA="true";  ASSET_NAME="asr-linux-aarch64-cuda" ;;
+                    2) USE_CUDA="false"; ASSET_NAME="asr-linux-aarch64" ;;
+                    *)
+                        warn "Invalid choice '${choice}', defaulting to CUDA."
+                        USE_CUDA="true"; ASSET_NAME="asr-linux-aarch64-cuda"
+                        ;;
+                esac
+            else
+                ASSET_NAME="asr-linux-aarch64"
+            fi
+            ;;
         macos-x86_64)
             err "macOS x86_64 (Intel) is not supported. Apple Silicon required."
             exit 1

skills/bootstrap.sh

@@ -33,6 +33,10 @@ detect_platform() {
     echo "${os}-${arch}"
 }
 
+has_nvidia_gpu() {
+    command -v nvidia-smi &>/dev/null && nvidia-smi --query-gpu=driver_version --format=csv,noheader &>/dev/null
+}
+
 get_asset_name() {
     local platform="$1"
 
@@ -41,7 +45,11 @@ get_asset_name() {
         echo "asr-linux-x86_64"
         ;;
     linux-aarch64)
-        echo "asr-linux-aarch64"
+        if has_nvidia_gpu; then
+            echo "asr-linux-aarch64-cuda"
+        else
+            echo "asr-linux-aarch64"
+        fi
         ;;
     darwin-aarch64)
         echo "asr-macos-aarch64"