File size: 1,969 Bytes
0914710
 
 
 
1ac44c5
0914710
 
 
9e9174e
0914710
b1bb633
 
 
 
 
 
 
 
 
d33fb86
b1bb633
 
0914710
 
 
 
 
 
 
9e9174e
0914710
9e9174e
 
0914710
b1bb633
0914710
b1bb633
0914710
 
 
 
 
9e9174e
1ac44c5
0914710
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
#!/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=${XDG_CACHE_HOME}/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