Skip to content
Draft
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions docker/Dockerfile.deepstream
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -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 && \
Expand Down Expand Up @@ -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 && \
Expand Down