change base image of grobid docker for delft

kermitt2 · Sep 10, 2021 · 436ffbc · 436ffbc
1 parent 98d9ed9
commit 436ffbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.delft b/Dockerfile.delft
@@ -61,7 +61,7 @@ RUN rm -rf grobid-source
 # -------------------
 
 # use NVIDIA Container Toolkit to automatically recognize possible GPU drivers on the host machine
-FROM nvidia/cuda:10.1-base
+FROM FROM tensorflow/tensorflow:1.15.5-gpu
 CMD nvidia-smi
 
 # setting locale is likely useless but to be sure