diff --git a/docker/tensorrt/Dockerfile.arm64 b/docker/tensorrt/Dockerfile.arm64 index 0ae9c38e9..215032166 100644 --- a/docker/tensorrt/Dockerfile.arm64 +++ b/docker/tensorrt/Dockerfile.arm64 @@ -112,7 +112,8 @@ RUN apt-get update \ && apt-get install -y protobuf-compiler libprotobuf-dev \ && rm -rf /var/lib/apt/lists/* RUN --mount=type=bind,source=docker/tensorrt/requirements-models-arm64.txt,target=/requirements-tensorrt-models.txt \ - pip3 wheel --wheel-dir=/trt-model-wheels -r /requirements-tensorrt-models.txt + --mount=type=bind,source=docker/tensorrt/requirements-arm64.txt,target=/requirements-tensorrt-constraints.txt \ + pip3 wheel --wheel-dir=/trt-model-wheels -c /requirements-tensorrt-constraints.txt -r /requirements-tensorrt-models.txt FROM wget AS jetson-ffmpeg ARG DEBIAN_FRONTEND diff --git a/docker/tensorrt/requirements-models-arm64.txt b/docker/tensorrt/requirements-models-arm64.txt index 4df74b128..fe89b4754 100644 --- a/docker/tensorrt/requirements-models-arm64.txt +++ b/docker/tensorrt/requirements-models-arm64.txt @@ -1,3 +1,2 @@ -numpy == 1.26.*; platform_machine == 'aarch64' onnx == 1.14.0; platform_machine == 'aarch64' protobuf == 3.20.3; platform_machine == 'aarch64'