2019-10-15 12:46:51 +02:00
|
|
|
#!/bin/sh
|
2020-09-05 12:59:40 +02:00
|
|
|
set -e
|
|
|
|
|
|
|
|
|
|
for tf in tf1 tf2; do
|
|
|
|
|
echo "== $tf"
|
|
|
|
|
case "$tf" in
|
|
|
|
|
tf1) BASE_IMAGE=nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04;;
|
|
|
|
|
tf2) BASE_IMAGE=nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04;;
|
|
|
|
|
esac
|
|
|
|
|
work_dir=`dirname $0`
|
|
|
|
|
image_id=`docker build -q --build-arg tf=$tf --build-arg BASE_IMAGE=$BASE_IMAGE -f assets/Dockerfile $work_dir`
|
2020-09-09 14:07:27 +02:00
|
|
|
docker run --gpus all -it --rm -e TF_CPP_MIN_LOG_LEVEL=1 $image_id
|
2020-09-05 12:59:40 +02:00
|
|
|
docker rmi $image_id >/dev/null || true
|
|
|
|
|
done
|
