run_jupyter.sh

#!/bin/bash

AIRT_DOCKER=ghcr.io/airtai/airt-docker-dask-tf2

BRANCH=$(git branch --show-current)
if [ "$BRANCH" == "main" ]
then
    TAG=latest
elif [ "$BRANCH" == "dev" ]
then
    TAG=dev
else
    if [ "$(docker image ls -q $AIRT_DOCKER:$BRANCH)" == "" ]
    then
        TAG=dev
    else
        TAG=$BRANCH
    fi
fi
export AIRT_DOCKER=$AIRT_DOCKER:$TAG



if test -z "$AIRT_JUPYTER_PORT"; then
      echo 'AIRT_JUPYTER_PORT variable not set, setting to 8888'
      export AIRT_JUPYTER_PORT=8888
else
      echo AIRT_JUPYTER_PORT variable set to $AIRT_JUPYTER_PORT
fi

if test -z "$AIRT_TB_PORT"; then
      echo 'AIRT_TB_PORT variable not set, setting to 6006'
      export AIRT_TB_PORT=6006
else
      echo AIRT_TB_PORT variable set to $AIRT_TB_PORT
fi

if test -z "$AIRT_DASK_PORT"; then
      echo 'AIRT_DASK_PORT variable not set, setting to 8787'
      export AIRT_DASK_PORT=8787
else
      echo AIRT_DASK_PORT variable set to $AIRT_DASK_PORT
fi

if test -z "$AIRT_DOCS_PORT"; then
      echo 'AIRT_DOCS_PORT variable not set, setting to 4000'
      export AIRT_DOCS_PORT=4000
else
      echo AIRT_DOCS_PORT variable set to $AIRT_DOCS_PORT
fi

if test -z "$AIRT_DATA"; then
      echo 'AIRT_DATA variable not set, setting to current directory'
      export AIRT_DATA=$(pwd)
fi
echo AIRT_DATA variable set to $AIRT_DATA

if test -z "$AIRT_PROJECT"; then
      echo 'AIRT_PROJECT variable not set, setting to current directory'
      export AIRT_PROJECT=$(pwd)
fi
echo AIRT_PROJECT variable set to $AIRT_PROJECT

echo Using $AIRT_DOCKER
docker image ls $AIRT_DOCKER

if $(which nvidia-smi); then
      echo INFO: Running docker image with all GPU-s
      nvidia-smi -L
      export GPU_PARAMS="--gpus all"
else
      echo INFO: Running docker image without GPU-s
      export GPU_PARAMS=""
fi

docker run -it --rm $GPU_PARAMS -u $(id -u):$(id -g) \
      -e JUPYTER_CONFIG_DIR=/root/.jupyter \
      -p $AIRT_JUPYTER_PORT:8888 -p $AIRT_TB_PORT:6006 -p $AIRT_DASK_PORT:8787 -p $AIRT_DOCS_PORT:4000 \
      -v $AIRT_DATA:/work/data -v $AIRT_PROJECT:/tf/airt \
      -v /etc/passwd:/etc/passwd -v /etc/group:/etc/group -v /etc/shadow:/etc/shadow \
      -v $HOME/.ssh:$HOME/.ssh -v $HOME/.gitconfig:/root/.gitconfig -v $HOME/.aws:/root/.aws \
      -e USER=$USER -e USERNAME=$USERNAME \
      $AIRT_DOCKER