chore: trt 10.5 (24.10 nvcr.io) CI + GPU dockers

jolibrain · Nov 4, 2024 · 69fcb04 · 69fcb04
1 parent 4071daf
commit 69fcb04
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/ci/devel-trt.Dockerfile b/ci/devel-trt.Dockerfile
@@ -1,19 +1,15 @@
 # syntax = docker/dockerfile:1.0-experimental
 
-ARG DD_UBUNTU_VERSION=22.04
-ARG DD_CUDA_VERSION=12.1
-ARG DD_CUDNN_VERSION=8
-ARG DD_TENSORRT_VERSION=8.6.1+cuda12.1
+ARG DD_CUDA_VERSION=12.6
+ARG DD_CUDNN_VERSION=9.5
+ARG DD_TENSORRT_VERSION=10.5.0+cuda12.6
 
-# FROM nvidia/cuda:${DD_CUDA_VERSION}-cudnn${DD_CUDNN_VERSION}-devel-ubuntu${DD_UBUNTU_VERSION}
-FROM nvcr.io/nvidia/tensorrt:23.05-py3 AS build
+FROM nvcr.io/nvidia/tensorrt:24.10-py3 AS build
 
-ARG DD_UBUNTU_VERSION
 ARG DD_CUDA_VERSION
 ARG DD_CUDNN_VERSION
 ARG DD_TENSORRT_VERSION
 
-RUN echo UBUNTU_VERSION=${DD_UBUNTU_VERSION} >> /image-info
 RUN echo CUDA_VERSION=${DD_CUDA_VERSION} >> /image-info
 RUN echo CUDNN_VERSION=${DD_CUDNN_VERSION} >> /image-info
 RUN echo TENSORRT_VERSION=${DD_TENSORRT_VERSION} >> /image-info

diff --git a/docker/gpu_tensorrt.Dockerfile b/docker/gpu_tensorrt.Dockerfile
@@ -1,5 +1,5 @@
 # syntax = docker/dockerfile:1.0-experimental
-FROM nvcr.io/nvidia/tensorrt:23.05-py3 AS build
+FROM nvcr.io/nvidia/tensorrt:24.10-py3 AS build
 
 ARG DEEPDETECT_RELEASE=OFF
 ARG DEEPDETECT_ARCH=gpu
@@ -144,7 +144,7 @@ RUN --mount=type=cache,target=/ccache/ mkdir build && cd build && ../build.sh
 RUN ./docker/get_libs.sh
 
 # Build final Docker image
-FROM nvcr.io/nvidia/tensorrt:23.05-py3 AS runtime
+FROM nvcr.io/nvidia/tensorrt:24.10-py3 AS runtime
 
 ARG DEEPDETECT_ARCH=gpu
 ARG DEEPDETECT_CUDA_VERSION=12.1