fix: wrong cuda runtime in docker images

Author: beniz

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
 

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"