fix: wrong cuda runtime in docker images

jolibrain · May 26, 2022 · 8ca5acf · 8ca5acf
1 parent fde90cd
commit 8ca5acf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/docker/gpu.Dockerfile b/docker/gpu.Dockerfile
@@ -111,7 +111,7 @@ RUN --mount=type=cache,target=/ccache/ mkdir build && cd build && ../build.sh
 RUN ./docker/get_libs.sh
 
 # Build final Docker image
-FROM nvidia/cuda:11.1-cudnn8-runtime-ubuntu20.04 AS runtime
+FROM nvidia/cuda:11.6.0-cudnn8-runtime-ubuntu20.04 AS runtime
 
 ARG DEEPDETECT_ARCH=gpu
 

diff --git a/docker/sync_gpu_dockerfiles.sh b/docker/sync_gpu_dockerfiles.sh
@@ -3,14 +3,14 @@
 here=$(dirname $(readlink -f $0))
 
 declare -A runtime_images
-runtime_images[gpu]="nvidia/cuda:11.1-cudnn8-runtime-ubuntu20.04"
+runtime_images[gpu]="nvidia/cuda:11.6.0-cudnn8-runtime-ubuntu20.04"
 # Ubuntu 20.04+cuda 11.1.80 + cuDNN 8 + TensorRT 7.2.x
 # https://docs.nvidia.com/deeplearning/tensorrt/container-release-notes/rel_20-12.html#rel_20-12
-runtime_images[gpu_tensorrt]="nvcr.io/nvidia/tensorrt:20.12-py3"
+runtime_images[gpu_tensorrt]="nvcr.io/nvidia/tensorrt:22.03-py3"
 
 declare -A devel_images
-devel_images[gpu]="nvidia/cuda:11.1-cudnn8-devel-ubuntu20.04"
-devel_images[gpu_tensorrt]="nvcr.io/nvidia/tensorrt:20.12-py3"
+devel_images[gpu]="nvidia/cuda:11.6.0-cudnn8-devel-ubuntu20.04"
+devel_images[gpu_tensorrt]="nvcr.io/nvidia/tensorrt:22.03-py3"
 
 declare -A builds
 builds[gpu_tensorrt]="tensorrt"