Skip to content

Commit d3cdf20

Browse files
committed
chore: trt 10.5 (24.10 nvcr.io) CI + GPU dockers
1 parent 808c4e1 commit d3cdf20

File tree

2 files changed

+6
-10
lines changed

2 files changed

+6
-10
lines changed

ci/devel-trt.Dockerfile

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,15 @@
11
# syntax = docker/dockerfile:1.0-experimental
22

3-
ARG DD_UBUNTU_VERSION=22.04
4-
ARG DD_CUDA_VERSION=12.1
5-
ARG DD_CUDNN_VERSION=8
6-
ARG DD_TENSORRT_VERSION=8.6.1+cuda12.1
3+
ARG DD_CUDA_VERSION=12.6
4+
ARG DD_CUDNN_VERSION=9.5
5+
ARG DD_TENSORRT_VERSION=10.5.0+cuda12.6
76

8-
# FROM nvidia/cuda:${DD_CUDA_VERSION}-cudnn${DD_CUDNN_VERSION}-devel-ubuntu${DD_UBUNTU_VERSION}
9-
FROM nvcr.io/nvidia/tensorrt:23.05-py3 AS build
7+
FROM nvcr.io/nvidia/tensorrt:24.10-py3 AS build
108

11-
ARG DD_UBUNTU_VERSION
129
ARG DD_CUDA_VERSION
1310
ARG DD_CUDNN_VERSION
1411
ARG DD_TENSORRT_VERSION
1512

16-
RUN echo UBUNTU_VERSION=${DD_UBUNTU_VERSION} >> /image-info
1713
RUN echo CUDA_VERSION=${DD_CUDA_VERSION} >> /image-info
1814
RUN echo CUDNN_VERSION=${DD_CUDNN_VERSION} >> /image-info
1915
RUN echo TENSORRT_VERSION=${DD_TENSORRT_VERSION} >> /image-info

docker/gpu_tensorrt.Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# syntax = docker/dockerfile:1.0-experimental
2-
FROM nvcr.io/nvidia/tensorrt:23.05-py3 AS build
2+
FROM nvcr.io/nvidia/tensorrt:24.10-py3 AS build
33

44
ARG DEEPDETECT_RELEASE=OFF
55
ARG DEEPDETECT_ARCH=gpu
@@ -144,7 +144,7 @@ RUN --mount=type=cache,target=/ccache/ mkdir build && cd build && ../build.sh
144144
RUN ./docker/get_libs.sh
145145

146146
# Build final Docker image
147-
FROM nvcr.io/nvidia/tensorrt:23.05-py3 AS runtime
147+
FROM nvcr.io/nvidia/tensorrt:24.10-py3 AS runtime
148148

149149
ARG DEEPDETECT_ARCH=gpu
150150
ARG DEEPDETECT_CUDA_VERSION=12.1

0 commit comments

Comments
 (0)