|
#!/usr/bin/env bash |
|
|
|
WORKDIR="/var/task" |
|
XDG_CACHE_HOME="/data" |
|
ROOT_PROJECT=${XDG_CACHE_HOME} |
|
MPLCONFIGDIR=${XDG_CACHE_HOME}/.cache/matplotlib |
|
TRANSFORMERS_CACHE=${XDG_CACHE_HOME}/.cache/transformers |
|
FASTAPI_STATIC=${XDG_CACHE_HOME}/static |
|
VIS_OUTPUT=${XDG_CACHE_HOME}/vis_output |
|
|
|
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 |
|
|
|
ls -ld ${XDG_CACHE_HOME}/ |
|
ls -l ${XDG_CACHE_HOME}/ |
|
export WORKDIR |
|
export XDG_CACHE_HOME |
|
export MPLCONFIGDIR |
|
export TRANSFORMERS_CACHE |
|
export FASTAPI_STATIC |
|
export VIS_OUTPUT |
|
export ROOT_PROJECT |
|
|
|
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 |
|
|
