diff --git a/docker/tensorrt/Dockerfile.amd64 b/docker/tensorrt/Dockerfile.amd64 index ef0295a96..cdf5df9ff 100644 --- a/docker/tensorrt/Dockerfile.amd64 +++ b/docker/tensorrt/Dockerfile.amd64 @@ -21,7 +21,7 @@ FROM deps AS frigate-tensorrt ARG PIP_BREAK_SYSTEM_PACKAGES RUN --mount=type=bind,from=trt-wheels,source=/trt-wheels,target=/deps/trt-wheels \ - pip3 uninstall -y onnxruntime tensorflow-cpu \ + pip3 uninstall -y onnxruntime \ && pip3 install -U /deps/trt-wheels/*.whl COPY --from=rootfs / /