26 lines
428 B
Bash
Executable File
26 lines
428 B
Bash
Executable File
#!/bin/bash
|
|
|
|
# Continue on error
|
|
set -e
|
|
|
|
# Build with progress output and capture the tag
|
|
TAG=$( \
|
|
docker build \
|
|
--target test \
|
|
. \
|
|
2>&1 | tee /dev/tty | grep "writing image" | cut -d' ' -f4 \
|
|
)
|
|
|
|
# Exit if tag is empty
|
|
[ -z "$TAG" ] && exit 1
|
|
|
|
# Run tests
|
|
docker run \
|
|
--rm \
|
|
--gpus=all \
|
|
--privileged \
|
|
-v /$(pwd)/model/:/root/model/ \
|
|
$TAG
|
|
|
|
# Clean up image
|
|
[ ! -z "$TAG" ] && docker rmi $TAG |