33 lines
716 B
Bash
Executable File
33 lines
716 B
Bash
Executable File
#!/bin/bash
|
|
|
|
export MSYS_NO_PATHCONV=1
|
|
set -e
|
|
|
|
function chown_iterations() {
|
|
if [ -d "./iterations" ] && [ "$(find ./iterations/ ! -user $USER -o ! -group $USER 2>/dev/null)" ]; then
|
|
echo "Chowning iterations directory"
|
|
sudo chown -R $USER:$USER ./iterations/
|
|
fi
|
|
}
|
|
|
|
trap chown_iterations EXIT
|
|
|
|
docker build \
|
|
--tag sia \
|
|
.
|
|
|
|
docker run \
|
|
--init \
|
|
--rm \
|
|
-ti \
|
|
--gpus=all \
|
|
-p 8080:8080 \
|
|
-e SIA_ITERATIONS_DIR=/root/iterations \
|
|
--env-file .env \
|
|
-v /$(pwd)/model/:/root/model/ \
|
|
-v /$(pwd)/.env:/root/.env \
|
|
-v /$(pwd)/iterations/:/root/iterations/ \
|
|
-v /$(pwd)/tools/:/root/sia/tools/ \
|
|
-v /$(pwd)/.git/:/root/sia/.git/ \
|
|
sia "$@"
|