mikegerber/test-nvidia
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

✨ Add run-docker-compatibility-matrix to test more combinations

Browse source
This commit is contained in:
Gerber, Mike 2020-09-09 14:07:27 +02:00
parent f0404f00ed
commit 7cb8eb75fa
3 changed files with 21 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

18
run-docker-compatibility-matrix Executable file
View file

@ -0,0 +1,18 @@
#!/bin/sh
set -e
for tf in tf1 tf2; do
# Note: CUDA 11.0 only with CUDNN 8
for BASE_IMAGE in \
nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04 \
nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 \
nvidia/cuda:10.2-cudnn7-runtime-ubuntu18.04 \
nvidia/cuda:11.0-cudnn8-runtime-ubuntu18.04 \
; do
echo "== $tf $BASE_IMAGE"
work_dir=`dirname $0`
image_id=`docker build -q --build-arg tf=$tf --build-arg BASE_IMAGE=$BASE_IMAGE -f assets/Dockerfile $work_dir`
docker run --gpus all -it --rm -e TF_CPP_MIN_LOG_LEVEL=2 $image_id
docker rmi $image_id >/dev/null || true
done
done