Fix docker build failure.

This commit is contained in:
Daniel
2025-08-13 15:30:37 +02:00
parent d14cd86742
commit a428a16c74

View File

@ -7,7 +7,7 @@ WORKDIR /
RUN apt-get update && apt-get install -y --no-install-recommends feh
RUN apt-get update && apt-get install -y --no-install-recommends python3-opencv
RUN pip uninstall -y opencv-python && pip install --no-cache "opencv-python<4.3"
RUN pip uninstall -y opencv-python && pip install --no-cache-dir "opencv-python<4.3"
# Show matplotlib images
RUN apt-get update && apt-get install -y --no-install-recommends python3-tk
@ -23,7 +23,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends libopencv-dev
# Install ONNX runtime
# See: https://github.com/microsoft/onnxruntime/blob/main/dockerfiles/Dockerfile.tensorrt
RUN pip3 uninstall -y onnxruntime-gpu
RUN pip3 install --no-cache-dir psutil
RUN git clone --recursive --depth=1 --branch=v1.20.1 https://github.com/Microsoft/onnxruntime.git
# Next line fixes: https://github.com/microsoft/onnxruntime/issues/24861
RUN cat /onnxruntime/cmake/deps.txt && \
sed -i 's/;be8be39fdbc6e60e94fa7870b280707069b5b81a/;32b145f525a8308d7ab1c09388b2e288312d8eba/g' /onnxruntime/cmake/deps.txt && \
cat /onnxruntime/cmake/deps.txt
ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/cmake-3.30.1-linux-x86_64/bin:${PATH}
ARG CMAKE_CUDA_ARCHITECTURES=75;80;90
ENV TRT_VERSION=10.5.0.18