scripts/entrypoint.sh

#!/usr/bin/env bash

WORKDIR="/var/task"
XDG_CACHE_HOME="/data"
PROJECT_ROOT_FOLDER=${XDG_CACHE_HOME}
MPLCONFIGDIR=${XDG_CACHE_HOME}/.cache/matplotlib
TRANSFORMERS_CACHE=${XDG_CACHE_HOME}/.cache/transformers
FASTAPI_STATIC=${WORKDIR}/static
VIS_OUTPUT=${XDG_CACHE_HOME}/vis_output

echo "WORKDIR content: ${WORKDIR}, before creating the FASTAPI_STATIC symlink ..."
ls -ld ${WORKDIR}
ls -lA ${WORKDIR}
cd ${WORKDIR}

ln -s ${FASTAPI_STATIC} .
echo "WORKDIR content: ${WORKDIR}, after creating the FASTAPI_STATIC symlink ..."
ls -ld ${WORKDIR}
ls -lA ${WORKDIR}
ls -lA ${WORKDIR}/static

echo "XDG_CACHE_HOME content: ${XDG_CACHE_HOME}, before creating the folders ..."
ls -ld ${XDG_CACHE_HOME}/
ls -l ${XDG_CACHE_HOME}/

mkdir -p ${XDG_CACHE_HOME}/.cache
mkdir -p ${MPLCONFIGDIR}
mkdir -p ${TRANSFORMERS_CACHE}
mkdir -p ${FASTAPI_STATIC}
mkdir -p ${VIS_OUTPUT}
chmod 770 -R ${FASTAPI_STATIC}
chmod 770 -R ${VIS_OUTPUT}
chmod 770 -R ${XDG_CACHE_HOME}/.cache

echo "XDG_CACHE_HOME content: ${XDG_CACHE_HOME}, after creating the folders ..."
ls -ld ${XDG_CACHE_HOME}/
ls -lA ${XDG_CACHE_HOME}/
export WORKDIR
export XDG_CACHE_HOME
export MPLCONFIGDIR
export TRANSFORMERS_CACHE
export FASTAPI_STATIC
export VIS_OUTPUT
export PROJECT_ROOT_FOLDER

source ${WORKDIR}/venv/bin/activate

which python
python --version

free -m
which nvcc
nvcc -V
which nvidia-smi
nvidia-smi
pip list

which uvicorn
ls -l ${WORKDIR}/venv/bin/uvicorn

df -h / /data /home /var/task

echo "WORKDIR - /var/task"
ls -l ${WORKDIR}

echo "XDG_CACHE_HOME - /data"
find ${XDG_CACHE_HOME}

CUDA_VISIBLE_DEVICES=$(nvidia-smi --query-gpu=memory.free,index --format=csv,nounits,noheader | sort -nr | head -1 | awk '{ print $NF }')
echo "calculated CUDA_VISIBLE_DEVICES env variable: ${CUDA_VISIBLE_DEVICES}."
export CUDA_VISIBLE_DEVICES

echo "running command 'uvicorn wrappers.fastapi_wrapper:app --host 0.0.0.0 --port 7860'..."
uvicorn wrappers.fastapi_wrapper:app --host 0.0.0.0 --port 7860

exit 0