diff --git a/docker/Dockerfile.deepstream b/docker/Dockerfile.deepstream index 342b75cb9..fb4749543 100644 --- a/docker/Dockerfile.deepstream +++ b/docker/Dockerfile.deepstream @@ -1,4 +1,4 @@ -ARG DEEPSTREAM_VERSION=7.0 +ARG DEEPSTREAM_VERSION=7.1 ARG DEEPSTREAM_DEVEL_IMAGE=$DEEPSTREAM_VERSION-triton-multiarch ARG DEEPSTREAM_BASE_IMAGE=$DEEPSTREAM_VERSION-samples-multiarch FROM nvcr.io/nvidia/deepstream:$DEEPSTREAM_DEVEL_IMAGE AS builder @@ -302,7 +302,7 @@ ARG TORCH2TRT_WHL="torch2trt-${TORCH2TRT_VERSION}-py3-none-any.whl" RUN wget -nv -P /usr/local/lib/ ${PACKAGES_URL}/libtorch2trt_plugins.so && \ ldconfig && \ wget -nv -O ${TORCH2TRT_WHL} ${PACKAGES_URL}/${TORCH2TRT_WHL} && \ - python -m pip install --no-cache-dir tensorrt~=8.6 && \ + python -m pip install --no-cache-dir tensorrt~=10.3 && \ python -m pip install --no-cache-dir ${TORCH2TRT_WHL} && \ python -m pip install --no-cache-dir nvidia-pyindex && \ python -m pip install --no-cache-dir onnx-graphsurgeon && \ @@ -374,7 +374,7 @@ ARG TORCH2TRT_WHL="torch2trt-${TORCH2TRT_VERSION}-py3-none-any.whl" RUN wget -nv -P /usr/local/lib/ ${PACKAGES_URL}/libtorch2trt_plugins.so && \ ldconfig && \ wget -nv -O ${TORCH2TRT_WHL} ${PACKAGES_URL}/${TORCH2TRT_WHL} && \ - python -m pip install --no-cache-dir tensorrt~=8.6 && \ + python -m pip install --no-cache-dir tensorrt~=10.3 && \ python -m pip install --no-cache-dir ${TORCH2TRT_WHL} && \ python -m pip install --no-cache-dir nvidia-pyindex && \ python -m pip install --no-cache-dir onnx-graphsurgeon && \