mirror of
https://github.com/jantic/DeOldify.git
synced 2026-08-30 18:02:24 +08:00
Jean-Louis Queguiner
52 lines
1.6 KiB
Bash
Executable File
52 lines
1.6 KiB
Bash
Executable File
#!/bin/bash
|
|
function usage {
|
|
echo '
|
|
_____ ____ _ _ _ __
|
|
| __ \ / __ \| | | (_)/ _|
|
|
| | | | ___| | | | | __| |_| |_ _ _
|
|
| | | |/ _ \ | | | |/ _` | | _| | | |
|
|
| |__| | __/ |__| | | (_| | | | | |_| |
|
|
|_____/ \___|\____/|_|\__,_|_|_| \__, |
|
|
__/ |
|
|
|___/
|
|
|
|
'
|
|
echo "usage : $0 notebook password -- to start the notebook with password"
|
|
echo " leave empty for no password (not recommended)"
|
|
echo "usage : $0 image_api -- to start image api"
|
|
echo "usage : $0 video_api -- to start video api"
|
|
echo ''
|
|
}
|
|
|
|
NOTEBOOK_PASSWORD=$2
|
|
|
|
if [ "$1" == "" ]; then
|
|
echo "missing first argument"
|
|
usage
|
|
else
|
|
#handling non GPU accelerated hardware
|
|
# if nvidia-smi not found => no ipc host
|
|
# --ipc=host
|
|
|
|
if nvidia-smi; then
|
|
IPC_HOST="--ipc=host"
|
|
echo "nvidia card found running on GPU"
|
|
else
|
|
IPC_HOST=""
|
|
echo "No nvidia card found running on CPU"
|
|
fi
|
|
|
|
already_built=$(rm -rf /tmp/deoldify.built; if docker image ls | grep deoldify > /tmp/deoldify.built; then echo "ok"; else echo "not ok"; fi; rm -rf /tmp/deoldify.built)
|
|
|
|
if [ "$already_built" == "ok" ]; then
|
|
echo "Docker started from cache"
|
|
echo "Access your $1 on port 5000 (api) or 8888 (notebook)"
|
|
docker run -it -p 8888:8888 -p 5000:5000 -e NOTEBOOK_PASSWORD=$NOTEBOOK_PASSWORD deoldify run_$1
|
|
else
|
|
docker build -t deoldify -f Dockerfile . && docker run -it -p 8888:8888 -p 5000:5000 -e NOTEBOOK_PASSWORD=$NOTEBOOK_PASSWORD deoldify run_$1
|
|
echo "Docker build and started"
|
|
echo "Access your $1 on port 5000 (api) or 8888 (notebook)"
|
|
fi
|
|
fi
|
|
|