run.sh

#!/usr/bin/env bash

cd "${BASH_SOURCE%/*}" || exit

if [ -z "$CONTAINER_NAME" ]; then
	CONTAINER_NAME="usgoc"
fi

if [ -z "$JUPYTER_PORT" ]; then
	JUPYTER_PORT=8888
fi

JUPYTER_TOKEN=${JUPYTER_TOKEN:-$(cat JUPYTER_TOKEN)}
JUPYTER_URL="http://localhost:$JUPYTER_PORT/?token=$JUPYTER_TOKEN"

if [ ! -z "$(docker ps -aqf "name=^$CONTAINER_NAME\$")" ]; then
	echo "Environment is already running in container ${CONTAINER_NAME}." >&2
	exit 1
fi

TF_FORCE_GPU_ALLOW_GROWTH=true
REBUILD="0"
ARGS="--rm -it"

if [ "$1" == "rebuild" ]; then
	REBUILD="1"

	if [ "$2" == "detached" ]; then
		ARGS="-d"
	fi
elif [ "$1" == "detached" ]; then
	ARGS="-d"

	if [ "$2" == "rebuild" ]; then
		REBUILD="1"
	fi
fi

if [[ "$(docker images -q usgoc/usgoc:latest 2> /dev/null)" == "" ]]; then
	REBUILD="1"
fi

if [ "$REBUILD" == "1" ]; then
	echo "Building container..."
	docker build . -t usgoc/usgoc -f Dockerfile
fi

echo "Starting Jupyter at ${JUPYTER_URL} ..."
echo ""

mkdir -p logs
mkdir -p evaluations
mkdir -p exported_models
mkdir -p data
mkdir -p plots
mkdir -p mlruns

docker run --gpus all --name $CONTAINER_NAME \
	-p $JUPYTER_PORT:8888 \
	-p 6006:6006 -p 10666:10666 -p 1234:1234 \
	-v $(pwd):/app \
	-v $(pwd)/data:/app/data -v $(pwd)/mlruns:/app/mlruns \
 	$ARGS \
	-u $(id -u):$(id -g) \
	--add-host=host.docker.internal:host-gateway \
	-e "JUPYTER_TOKEN=$JUPYTER_TOKEN" \
	-e TF_FORCE_GPU_ALLOW_GROWTH=$TF_FORCE_GPU_ALLOW_GROWTH \
	-e "USER=$USER" \
	-e "PROJECT_ROOT=/app" \
	usgoc/usgoc