From 5e20d251d67390e9843254eab95383ac091212f2 Mon Sep 17 00:00:00 2001
From: "Gerber, Mike" <mike.gerber@sbb.spk-berlin.de>
Date: Mon, 1 Feb 2021 18:19:04 +0100
Subject: [PATCH] =?UTF-8?q?=E2=AC=86=EF=B8=8F=20Update=20base=20image=20fo?=
 =?UTF-8?q?r=20TF2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 run-docker | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/run-docker b/run-docker
index af5c448..584b4c3 100755
--- a/run-docker
+++ b/run-docker
@@ -5,7 +5,7 @@ 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;;
+    tf2) BASE_IMAGE=nvidia/cuda:11.0-cudnn8-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`