17 lines
284 B
Bash
Executable File
17 lines
284 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -e
|
|
set -x
|
|
echo "Running in $(pwd)"
|
|
|
|
|
|
case $IMAGE_NAME in
|
|
*-cuda)
|
|
docker build --build-arg BASE_IMAGE=nvidia/cuda:10.1-runtime-ubi8 -f $(basename $DOCKERFILE_PATH) -t $IMAGE_NAME ..
|
|
;;
|
|
|
|
*)
|
|
docker build -f $(basename $DOCKERFILE_PATH) -t $IMAGE_NAME ..
|
|
;;
|
|
esac
|