fix(docker): fixup base runtime image

jolibrain · Oct 7, 2020 · 6238dd4 · 6238dd4
1 parent 0f05495
commit 6238dd4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/docker/gpu.Dockerfile b/docker/gpu.Dockerfile
@@ -107,7 +107,7 @@ RUN --mount=type=cache,target=/ccache/ mkdir build && cd build && ../build.sh
 RUN ./docker/get_libs.sh
 
 # Build final Docker image
-FROM ubuntu:18.04 AS runtime
+FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 AS runtime
 
 ARG DEEPDETECT_ARCH=gpu
 

diff --git a/docker/gpu_tensorrt.Dockerfile b/docker/gpu_tensorrt.Dockerfile
@@ -107,7 +107,7 @@ RUN --mount=type=cache,target=/ccache/ mkdir build && cd build && ../build.sh
 RUN ./docker/get_libs.sh
 
 # Build final Docker image
-FROM ubuntu:18.04 AS runtime
+FROM nvcr.io/nvidia/tensorrt:20.03-py3 AS runtime
 
 ARG DEEPDETECT_ARCH=gpu
 
@@ -122,6 +122,7 @@ LABEL maintainer="[email protected]"
 RUN --mount=type=cache,target=/var/cache/apt --mount=type=cache,target=/var/lib/apt \
     apt-get update -y && apt-get install -y \
     wget \
+    curl \
 	libopenblas-base \
 	liblmdb0 \
 	libleveldb1v5 \

diff --git a/docker/sync_gpu_dockerfiles.sh b/docker/sync_gpu_dockerfiles.sh
@@ -11,7 +11,7 @@ for dest in "${!images[@]}" ; do
     image=${images[$dest]}
 
     sed \
-        -e "s,FROM [^ ]* AS build,FROM ${image} AS build,g" \
+        -e "s,FROM [^ ]*,FROM ${image},g" \
         -e "s,ARG DEEPDETECT_ARCH=.*,ARG DEEPDETECT_ARCH=gpu,g" \
         cpu.Dockerfile > ${dest}.Dockerfile
 done