setup_runner_pc.sh

#!/usr/bin/env bash

# Copyright (c) 2018-2025, Texas Instruments
# All Rights Reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice, this
#   list of conditions and the following disclaimer.
#
# * Redistributions in binary form must reproduce the above copyright notice,
#   this list of conditions and the following disclaimer in the documentation
#   and/or other materials provided with the distribution.
#
# * Neither the name of the copyright holder nor the names of its
#   contributors may be used to endorse or promote products derived from
#   this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.


######################################################################
# change the default here if needed - supported options: 11.2 11.1 11.0 10.1 10.0
TIDL_TOOLS_VERSION=${TIDL_TOOLS_VERSION-"11.2"}

# change the default here if needed - supported options: cpu gpu
# if you are invoking the script setup_runner_pc_gpu.sh, 
# then you would need to change this in that script and not here.
TIDL_TOOLS_TYPE=${TIDL_TOOLS_TYPE-"cpu"}

######################################################################
CURRENT_WORK_DIR=$(pwd)


#######################################################################
echo "INFO: setting up TIDL runner environment in PC with TIDL_TOOLS_VERSION=${TIDL_TOOLS_VERSION} and TIDL_TOOLS_TYPE=${TIDL_TOOLS_TYPE}"
echo "INFO: TIDL_TOOLS_VERSION can be changed in setup_runner_pc.sh if needed."

sleep 5

echo 'INFO: installing system dependencies...'

# Function to check if a package is installed using dpkg
is_package_installed() {
    local package_name="$1"
    if dpkg-query -W -f='${Status}' "$package_name" 2>/dev/null | grep -q "install ok installed"; then
        return 0  # Package is installed
    else
        return 1  # Package is not installed
    fi
}

# Function to install package only if not already installed
install_if_missing() {
    local package_name="$1"
    if is_package_installed "$package_name"; then
        echo "INFO: ✓ Package $package_name is already installed"
    else
        echo "INFO: installing $package_name..."
        sudo apt-get install -y "$package_name"
    fi
}

# Dependencies for cmake, onnx, pillow-simd, tidl-graph-visualization
# TBD: "graphviz-dev"
packages=("cmake" "libffi-dev" "libjpeg-dev" "zlib1g-dev" "protobuf-compiler" "graphviz")

for package in "${packages[@]}"; do
    install_if_missing "$package"
done


#######################################################################
pip3 install -e ./tools


######################################################################
# unsintall onnxruntime and install onnxruntime-tild along with tidl-tools
# pip3 uninstall -y onnxruntime onnxruntime-tidl


# tidlrunner-tools-download is a script that defined in and installed via tools/pyproject.toml
# tidlrunner-tools-download - this invokes: python3 tools/tidl_tools_package/download.py
echo "INFO: running: tidlrunner-tools-download..."
TIDL_TOOLS_TYPE=${TIDL_TOOLS_TYPE} TIDL_TOOLS_VERSION=${TIDL_TOOLS_VERSION} tidlrunner-tools-download


######################################################################
echo "INFO: installing edgeai_tidirunner package..."
pip3 install -e ./tidlrunner[pc]


######################################################################
# tidlrunner-tools-install is a script that defined in and installed via ./pyproject.toml
# it installs additional python packages and tools
echo "Running: tidlrunner-tools-install..."
tidlrunner-tools-install