#!/bin/bash

# Set CUDA environment
if [ -d "/usr/local/cuda" ]; then
    export CUDA_HOME=/usr/local/cuda
    echo "CUDA environment set to $CUDA_HOME"
elif [ -d "/usr/lib/cuda" ]; then
    export CUDA_HOME=/usr/lib/cuda
    echo "CUDA environment set to $CUDA_HOME"
else
    echo "CUDA not found. Running in CPU-only mode."
fi

# Set LibTorch library path
if [ -d "$HOME/libtorch" ]; then
    export LD_LIBRARY_PATH=$HOME/libtorch/lib:$LD_LIBRARY_PATH
    echo "LibTorch library path set to $HOME/libtorch/lib"
elif [ -d "/usr/local/libtorch" ]; then
    export LD_LIBRARY_PATH=/usr/local/libtorch/lib:$LD_LIBRARY_PATH
    echo "LibTorch library path set to /usr/local/libtorch/lib"
fi

# Run the demo
echo "Starting tracking demo..."
./bin/tracking_demo