install_minimal.sh

#!/bin/bash

sudo apt-get install jq

json_path='docker_sdk_api/api/data/paths.json'

# change training api image name based on architecture CPU/GPU
change_image_name(){
jq '."image_name"="'$1'"'  "$json_path" >temp.$$.json && mv temp.$$.json "$json_path"
}

adjust_base_dir(){

    #Adjust basedir path
    python3 adjust_basedir_path.py
}

CPU_docker_image="classification_training_api_cpu"
GPU_docker_image="classification_training_api_gpu"
echo '----------------------------------------------'
echo 'Please choose docker image build architecture'

PS3='Please enter your choice: '
options=("GPU" "CPU")

adjust_base_dir

select opt in "${options[@]}"
do
    case $opt in
        "GPU")
            change_image_name "$GPU_docker_image"
            break
            ;;
        "CPU")
            
            change_image_name "$CPU_docker_image"
            break
            ;;
        *) echo "invalid option $REPLY";;
    esac
done