40 lines
890 B
Bash
Executable File
40 lines
890 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
|
|
}
|
|
|
|
function clean_docker() {
|
|
docker images -f "dangling=true" -q | xargs -r docker rmi
|
|
}
|
|
|
|
trap chown_iterations EXIT
|
|
trap clean_docker EXIT
|
|
|
|
docker build \
|
|
--tag sia \
|
|
.
|
|
|
|
docker run \
|
|
--init \
|
|
--rm \
|
|
-ti \
|
|
--gpus=all \
|
|
-p 8080:8080 \
|
|
--env-file .env \
|
|
-v /$(pwd)/models/:/root/models/ \
|
|
-v /$(pwd)/iterations/:/root/data/iterations/ \
|
|
-v /$(pwd)/tasks/:/root/data/tasks/ \
|
|
-v /$(pwd)/user/:/root/data/user/ \
|
|
-v /$(pwd)/environment/:/root/data/environment/ \
|
|
-v /$(pwd)/cache/:/root/.cache/ \
|
|
-v /$(pwd)/:/root/sia/ \
|
|
sia "$@"
|
|
|
|
exit $? |