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

✨ Test TensorFlow 2

Browse source
This commit is contained in:
Gerber, Mike 2020-09-05 12:59:40 +02:00
parent cfeafeb57f
commit 5fa9406fbd
8 changed files with 68 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

15
run-docker
View file

@ -1,3 +1,14 @@
#!/bin/sh
docker build -t test-nvidia `dirname $0`
docker run --gpus all -it --rm test-nvidia
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`
docker run --gpus all -it --rm $image_id
docker rmi $image_id >/dev/null || true
done