Use constraint instead

This commit is contained in:
Nicolas Mowen 2025-11-05 06:02:51 -07:00
parent 32b2d3afdd
commit d94584239a
2 changed files with 2 additions and 2 deletions

View File

@ -112,7 +112,8 @@ RUN apt-get update \
&& apt-get install -y protobuf-compiler libprotobuf-dev \ && apt-get install -y protobuf-compiler libprotobuf-dev \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
RUN --mount=type=bind,source=docker/tensorrt/requirements-models-arm64.txt,target=/requirements-tensorrt-models.txt \ 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 FROM wget AS jetson-ffmpeg
ARG DEBIAN_FRONTEND ARG DEBIAN_FRONTEND

View File

@ -1,3 +1,2 @@
numpy == 1.26.*; platform_machine == 'aarch64'
onnx == 1.14.0; platform_machine == 'aarch64' onnx == 1.14.0; platform_machine == 'aarch64'
protobuf == 3.20.3; platform_machine == 'aarch64' protobuf == 3.20.3; platform_machine == 'aarch64'