Skip to content

Commit

Permalink
Downgrade tensorrt version amd64 dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
weitheng01 committed Dec 23, 2024
1 parent 2d7e9c8 commit 52ee4b0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docker/tensorrt/Dockerfile.amd64
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ COPY docker/tensorrt/requirements-amd64.txt /requirements-tensorrt.txt
RUN mkdir -p /trt-wheels && pip3 wheel --wheel-dir=/trt-wheels -r /requirements-tensorrt.txt

FROM tensorrt-base AS frigate-tensorrt
ENV TRT_VER=8.6.1
ENV TRT_VER=8.5.3
RUN python3 -m pip config set global.break-system-packages true
RUN --mount=type=bind,from=trt-wheels,source=/trt-wheels,target=/deps/trt-wheels \
pip3 install -U /deps/trt-wheels/*.whl --break-system-packages && \
Expand Down

0 comments on commit 52ee4b0

Please sign in to comment.