2019-08-30 18:35:10 +00:00
|
|
|
#!/bin/bash
|
|
|
|
|
|
2019-08-30 19:37:42 +00:00
|
|
|
set -e
|
|
|
|
|
set -x
|
|
|
|
|
echo "Running in $(pwd)"
|
|
|
|
|
|
2019-08-30 18:35:10 +00:00
|
|
|
case $DOCKER_TAG in
|
|
|
|
|
*-cuda)
|
|
|
|
|
docker build --build-arg BASE_IMAGE=cuda:10.1-runtime-ubi7 -f $DOCKERFILE_PATH -t $IMAGE_NAME .
|
|
|
|
|
;;
|
|
|
|
|
|
|
|
|
|
*)
|
|
|
|
|
docker build -f $DOCKERFILE_PATH -t $IMAGE_NAME .
|
|
|
|
|
;;
|
|
|
|
|
esac
|
