#!/bin/bash
set -e
set -x
echo "Running in $(pwd)"
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